test updates

2026-02-21 11:38:06 -06:00
parent d71d40c4d3
commit 81793a7bfe
4 changed files with 12 additions and 15 deletions
--- a/tests/test_dag.py
+++ b/tests/test_dag.py
@@ -25,8 +25,9 @@ class TestDAGOps:
        def test_add_single_edge(self) -> None:
            g = DAG[str]()
            g.add_edge("a", "b")
-            assert "b" in g["a"]
-            assert "a" in g.reverse["b"]
+            assert "a" in g._succ
+            assert "b" in g._succ["a"]
+            assert "a" in g._pred["b"]

        def test_add_multiple_edges(self) -> None:
            g = DAG[str]()