Mypy

neo4j · Mar 4, 2025 · b8222a1 · b8222a1
1 parent 67a2481
commit b8222a1
Showing 1 changed file with 4 additions and 17 deletions.
diff --git a/tests/unit/experimental/pipeline/test_orchestrator.py b/tests/unit/experimental/pipeline/test_orchestrator.py
@@ -181,12 +181,12 @@ async def test_orchestrator_set_component_status(mock_status: Mock) -> None:
     mock_status.side_effect = [
         RunStatus.UNKNOWN,
     ]
-    assert await orchestrator.set_task_status("task_name", RunStatus.RUNNING) is None
+    await orchestrator.set_task_status("task_name", RunStatus.RUNNING)
     # RUNNING -> DONE
     mock_status.side_effect = [
         RunStatus.RUNNING,
     ]
-    assert await orchestrator.set_task_status("task_name", RunStatus.DONE) is None
+    await orchestrator.set_task_status("task_name", RunStatus.DONE)
     # Error path, raising PipelineStatusUpdateError
     # Same status
     # RUNNING -> RUNNING
@@ -238,30 +238,17 @@ async def test_orchestrator_check_dependency_complete(
     """a -> b, c"""
     orchestrator = Orchestrator(pipeline=pipeline_branch)
     node_a = pipeline_branch.get_node_by_name("a")
-    assert await orchestrator.check_dependencies_complete(node_a) is None
+    await orchestrator.check_dependencies_complete(node_a)
     node_b = pipeline_branch.get_node_by_name("b")
     # dependency is DONE:
     mock_status.side_effect = [RunStatus.DONE]
-    assert await orchestrator.check_dependencies_complete(node_b) is None
+    await orchestrator.check_dependencies_complete(node_b)
     # dependency is not DONE:
     mock_status.side_effect = [RunStatus.RUNNING]
     with pytest.raises(PipelineMissingDependencyError):
         await orchestrator.check_dependencies_complete(node_b)
 
 
-@pytest.mark.asyncio
-@patch(
-    "neo4j_graphrag.experimental.pipeline.pipeline.Orchestrator.get_status_for_component"
-)
-async def test_orchestrator_check_dependency_complete(
-    mock_status: Mock, pipeline_branch: Pipeline
-) -> None:
-    """a -> b, c"""
-    orchestrator = Orchestrator(pipeline=pipeline_branch)
-    node_a = pipeline_branch.get_node_by_name("a")
-    assert await orchestrator.check_dependencies_complete(node_a) is None
-
-
 @pytest.mark.asyncio
 @patch(
     "neo4j_graphrag.experimental.pipeline.pipeline.Orchestrator.get_status_for_component"