labnotes 0.0.135__tar.gz → 0.0.136__tar.gz

{labnotes-0.0.135 → labnotes-0.0.136}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: labnotes
-Version: 0.0.135
+Version: 0.0.136
 Summary: Browse experiment designs, notes, and run data from a folder.
 Project-URL: Repository, https://github.com/thekevinscott/labnotes
 Author-email: Kevin Scott <me@thekevinscott.com>

labnotes-0.0.136/tests/conftest.py

@@ -0,0 +1,76 @@
+import shlex
+import shutil
+import subprocess
+import sys
+from collections.abc import Iterator
+from pathlib import Path
+
+import pytest
+
+BOTBOX_CONTAINER = """
+import os
+import subprocess
+from pathlib import Path
+
+folder = Path(os.environ["LABNOTES_RUN"])
+labels = folder / "botbox-labels.txt"
+labels.write_text(os.environ["BOTBOX_LABELS"])
+subprocess.run([
+    "docker", "run", "--rm", "--label-file", str(labels),
+    "--cidfile", str(folder / "container.cid"),
+    "python:3.12-slim", "python", "-c",
+    "held = b'x' * 200_000_000; print(len(held), flush=True); "
+    "import time; time.sleep(3)",
+], check=True)
+"""
+
+HOOK = """
+import os
+import sys
+from pathlib import Path
+
+Path(sys.argv[1] + "-labels.txt").write_text(os.environ["BOTBOX_LABELS"])
+"""
+
+
+@pytest.fixture
+def botbox_caller_labels() -> str:
+    return "team=research\npurpose=botbox-memory-proof"
+
+
+@pytest.fixture
+def botbox_memory_project(tmp_path: Path) -> Iterator[Path]:
+    if shutil.which("docker") is None:
+        pytest.skip("docker is unavailable")
+    try:
+        available = subprocess.run(
+            ["docker", "info"], capture_output=True, timeout=3, check=False
+        )
+    except (OSError, subprocess.TimeoutExpired):
+        pytest.skip("docker daemon is unavailable")
+    if available.returncode != 0:
+        pytest.skip("docker daemon is unavailable")
+
+    root = tmp_path / "botbox memory project"
+    (root / "exp").mkdir(parents=True)
+    (root / ".labnotes.toml").write_text("")
+    (root / "script.py").write_text(BOTBOX_CONTAINER)
+    (root / "hook.py").write_text(HOOK)
+    python = shlex.quote(sys.executable)
+    (root / "exp" / "experiment.toml").write_text(
+        f'command = "{python} script.py"\n'
+        f'setup = "{python} hook.py setup"\n'
+        f'teardown = "{python} hook.py teardown"\n'
+        "memory_interval = 0.2\n"
+    )
+    try:
+        yield root
+    finally:
+        # --rm handles normal exits; remove our container if a test times out.
+        for cidfile in (root / "exp" / "runs").rglob("container.cid"):
+            subprocess.run(
+                ["docker", "rm", "--force", cidfile.read_text().strip()],
+                capture_output=True,
+                timeout=10,
+                check=False,
+            )

labnotes-0.0.136/tests/e2e/conftest.py

@@ -0,0 +1,28 @@
+import os
+import subprocess
+from pathlib import Path
+
+import pytest
+
+PACKAGE = Path(__file__).parents[2]
+
+
+@pytest.fixture
+def uvx(tmp_path: Path) -> dict[str, str]:
+    # Install this checkout as the uv tool so uvx runs it offline.
+    env = {"UV_NO_CONFIG": "1", "UV_TOOL_DIR": str(tmp_path / "tools")}
+    constraints = tmp_path / "constraints.txt"
+    uv = ["uv", "--directory", str(PACKAGE)]
+    subprocess.run(
+        [*uv, "export", "-q", "--frozen", "--no-dev", "--no-emit-project"]
+        + ["-o", str(constraints)],
+        check=True,
+    )
+    # Rebuild labnotes so uncommitted source changes are tested, too.
+    subprocess.run(
+        [*uv, "tool", "install", "-q", "--refresh-package", "labnotes"]
+        + ["-c", str(constraints), "."],
+        check=True,
+        env={**os.environ, **env, "UV_TOOL_BIN_DIR": str(tmp_path / "bin")},
+    )
+    return {**env, "UV_OFFLINE": "1"}