Skip to content

Commit 4d5382a

Browse files
reset current egraph after tests
1 parent f214e23 commit 4d5382a

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

python/tests/conftest.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
from syrupy.extensions.single_file import SingleFileSnapshotExtension
55

66
import egglog.conversion
7+
import egglog.exp.array_api
78

89

910
@pytest.fixture(autouse=True)
@@ -15,6 +16,12 @@ def _reset_conversions():
1516
egglog.conversion._TO_PROCESS_DECLS = old_conversion_decls
1617

1718

19+
@pytest.fixture(autouse=True)
20+
def _reset_current_egraph():
21+
yield
22+
egglog.exp.array_api._CURRENT_EGRAPH = None
23+
24+
1825
class PythonSnapshotExtension(SingleFileSnapshotExtension):
1926
_file_extension = "py"
2027

0 commit comments

Comments
 (0)