diff --git a/tests/test_assoccomm.py b/tests/test_assoccomm.py index 3795701..ce3362c 100644 --- a/tests/test_assoccomm.py +++ b/tests/test_assoccomm.py @@ -83,8 +83,6 @@ def results(g, s=None): def test_eq_comm(): x, y, z = var(), var(), var() - commutative.facts.clear() - commutative.index.clear() comm_op = "comm_op" @@ -326,8 +324,6 @@ def test_eq_assoc(): assoc_op = "assoc_op" - associative.index.clear() - associative.facts.clear() fact(associative, assoc_op) @@ -449,8 +445,6 @@ def test_eq_assoccomm(): ac = "commassoc_op" - commutative.index.clear() - commutative.facts.clear() fact(commutative, ac) fact(associative, ac) @@ -545,10 +539,6 @@ def test_assoccomm_algebra(): add = "add" mul = "mul" - commutative.index.clear() - commutative.facts.clear() - associative.index.clear() - associative.facts.clear() fact(commutative, add) fact(associative, add) @@ -565,10 +555,6 @@ def test_assoccomm_algebra(): def test_assoccomm_objects(): - commutative.index.clear() - commutative.facts.clear() - associative.index.clear() - associative.facts.clear() fact(commutative, Add) fact(associative, Add) @@ -578,3 +564,11 @@ def test_assoccomm_objects(): assert run(0, True, eq_assoccomm(add(1, 2, 3), add(3, 1, 2))) == (True,) assert run(0, x, eq_assoccomm(add(1, 2, 3), add(1, 2, x))) == (3,) assert run(0, x, eq_assoccomm(add(1, 2, 3), add(x, 2, 1))) == (3,) + +@pytest.fixture(autouse=True) +def clear_assoccomm(): + commutative.index.clear() + commutative.facts.clear() + associative.index.clear() + associative.facts.clear() + yield