Fix coverage

Author: dmontagu

Makefile

@@ -34,12 +34,11 @@ lint: ## Lint the code
 	uv run ruff format --check
 	uv run ruff check
 
-PYRIGHT_PYTHON ?= 3.10
-
+PYRIGHT_PYTHON ?= 3.13
 .PHONY: typecheck-pyright
 typecheck-pyright:
 	@# PYRIGHT_PYTHON_IGNORE_WARNINGS avoids the overhead of making a request to github on every invocation
-	PYRIGHT_PYTHON_IGNORE_WARNINGS=1 uv run pyright --pythonversion $(PYRIGHT_PYTHON)
+	PYRIGHT_PYTHON_IGNORE_WARNINGS=1 UV_PROJECT_ENVIRONMENT=.venv$(subst .,,$(PYRIGHT_PYTHON)) uv run pyright --pythonversion $(PYRIGHT_PYTHON)
 
 .PHONY: typecheck-mypy
 typecheck-mypy:
@@ -66,6 +65,13 @@ test-all-python: ## Run tests on Python 3.10 to 3.13
 	@uv run coverage combine
 	@uv run coverage report
 
+PYTEST_PYTHON ?= 3.13
+.PHONY: test-specific-python
+test-specific-python: ## Run tests and collect coverage data using a specific python, specified by PYTEST_PYTHON env var
+	UV_PROJECT_ENVIRONMENT=.venv$(subst .,,$(PYTEST_PYTHON)) uv run --python $(PYTEST_PYTHON) --all-extras --all-packages coverage run -p -m pytest -n auto --dist=loadgroup --durations=20
+	@uv run coverage combine
+	@uv run coverage report
+
 .PHONY: testcov
 testcov: test ## Run tests and generate an HTML coverage report
 	@echo "building coverage html"

tests/graph/beta/test_graph_iteration.py

@@ -238,7 +238,7 @@ async def step_three(ctx: StepContext[IterState, None, int]) -> int:  # pragma:
 
     async with graph.iter(state=state) as run:
         event_count = 0
-        async for _ in run:
+        async for _ in run:  # pragma: no branch
             event_count += 1
             if event_count >= 2:
                 break  # Early termination