test_stopwatch uses freezegun (#4138)

Use freezegun instead of `time.sleep`.

Author: berekuk

.github/actions/setup-environment/action.yml

@@ -122,7 +122,7 @@ runs:
 
           "testing")
             # Install main dependencies + testing group, excluding interactive and dev
-            uv sync --no-group interactive --no-group dev --group testing
+            uv sync --no-group interactive --no-group dev --group testing --all-packages
             ;;
 
           # used in the mettascope tests where we run code to generate outputs but do not run tests

metta/setup/components/uv.py

@@ -27,7 +27,7 @@ def check_installed(self) -> bool:
             sys.exit(1)
 
     def install(self, non_interactive: bool = False, force: bool = False) -> None:
-        cmd = ["uv", "sync"]
+        cmd = ["uv", "sync", "--all-packages"]
         cmd.extend(["--force-reinstall", "--no-cache"] if force else [])
         env = os.environ.copy()
         env["METTAGRID_FORCE_NIM_BUILD"] = "1"

packages/mettagrid/pyproject.toml

@@ -30,7 +30,8 @@ dependencies = [
 ]
 
 [dependency-groups]
-dev = ["pytest>=8.3.3", "pytest-benchmark>=5.1.0", "pytest-xdist>=3.8.0"]
+testing = ["pytest>=8.3.3", "pytest-benchmark>=5.1.0", "pytest-xdist>=3.8.0", "freezegun>=1.5.0"]
+dev = [{ include-group = "testing" }]
 
 [tool.bazel]
 # Bazel build configuration

packages/mettagrid/python/src/mettagrid/profiling/stopwatch.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import functools
 import inspect
 import logging
@@ -81,7 +83,7 @@ def _capture_caller_info(extra_skip_frames: int = 0) -> Tuple[str, int]:
         del frame
 
 
-def with_timer(timer: "Stopwatch", name: str, log_level: int | None = None):
+def with_timer(timer: Stopwatch, name: str, log_level: int | None = None):
     """Decorator that wraps function execution in a timer context."""
 
     def decorator(func: F) -> F:
@@ -288,7 +290,7 @@ def time(
                 display_name = name or self.GLOBAL_TIMER_NAME
                 self.logger.log(log_level, f"{display_name} took {elapsed:.3f}s")
 
-    def __call__(self, name: str | None = None, log_level: int | None = None) -> ContextManager["Stopwatch"]:
+    def __call__(self, name: str | None = None, log_level: int | None = None) -> ContextManager[Stopwatch]:
         """Make Stopwatch callable to return context manager.
 
         Args: