Fix test

stellasia · stellasia · commit ca445669a6d8 · 2025-09-01T14:05:51.000+02:00
diff --git a/tests/unit/experimental/utils/test_schema.py b/tests/unit/experimental/utils/test_schema.py
@@ -77,7 +77,7 @@ def invalid_schema_dict() -> dict[str, Any]:
     }
 
 
-@patch("neo4j_graphrag.experimental.utils.schema.neo4j_viz", None)
+@patch("neo4j_graphrag.experimental.utils.schema.VisualizationGraph", None)
 def test_schema_visualization_import_error() -> None:
     with pytest.raises(ImportError):
         schema_visualization({})

Original file line number	Diff line number	Diff line change
`@@ -77,7 +77,7 @@ def invalid_schema_dict() -> dict[str, Any]:`
`77`	`77`	`}`
`78`	`78`
`79`	`79`
`80`		`-@patch("neo4j_graphrag.experimental.utils.schema.neo4j_viz", None)`
	`80`	`+@patch("neo4j_graphrag.experimental.utils.schema.VisualizationGraph", None)`
`81`	`81`	`def test_schema_visualization_import_error() -> None:`
`82`	`82`	`with pytest.raises(ImportError):`
`83`	`83`	`schema_visualization({})`