fix test

opencybersecurityalliance · Jul 28, 2024 · a597b4f · a597b4f
1 parent a207a94
commit a597b4f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/packages/kestrel_core/tests/test_session.py b/packages/kestrel_core/tests/test_session.py
@@ -10,7 +10,7 @@
 from kestrel.config.internal import CACHE_INTERFACE_IDENTIFIER
 from kestrel.display import GraphExplanation
 from kestrel.frontend.parser import parse_kestrel_and_update_irgraph
-from kestrel.ir.graph import IRGraph
+from kestrel.ir.graph import IRGraph, IRGraphEvaluable
 from kestrel.ir.instructions import Construct, SerializableDataFrame
 
 
@@ -200,7 +200,7 @@ def test_explain_in_cache():
         assert isinstance(res, GraphExplanation)
         assert len(res.graphlets) == 1
         ge = res.graphlets[0]
-        assert ge.graph == session.irgraph.to_dict()
+        assert ge.graph == IRGraphEvaluable(session.irgraph).to_dict()
         construct = session.irgraph.get_nodes_by_type(Construct)[0]
         assert ge.action.language == "SQL"
         stmt = ge.action.statement.replace('"', '')