Remove repr full checks

OpenFreeEnergy · Feb 5, 2025 · 6bd6e71 · 6bd6e71
1 parent 2af7007
commit 6bd6e71
Showing 1 changed file with 18 additions and 4 deletions.
diff --git a/src/pontibus/tests/protocols/solvation/test_tokenization.py b/src/pontibus/tests/protocols/solvation/test_tokenization.py
@@ -61,13 +61,20 @@ def protocol_result(afe_solv_transformation_json):
 
 class TestProtocol(GufeTokenizableTestsMixin):
     cls = ASFEProtocol
-    key = "ASFEProtocol-6d821918a2d558cda0c5bfa7d6d5bace"
-    repr = f"<{key}>"
+    key = None
+    repr = "ASFEProtocol-"
 
     @pytest.fixture()
     def instance(self, protocol):
         return protocol
 
+    def test_repr(self, instance):
+        """
+        Overwrites the base `test_repr` call.
+        """
+        assert isinstance(repr(instance), str)
+        assert self.repr in repr(instance)
+
 
 class TestSolventUnit(GufeTokenizableTestsMixin):
     cls = ASFESolventUnit
@@ -111,9 +118,16 @@ def test_repr(self, instance):
 
 class TestProtocolResult(GufeTokenizableTestsMixin):
     cls = ASFEProtocolResult
-    key = "ASFEProtocolResult-e711f21656c3795ed9d545c326ec717a"
-    repr = f"<{key}>"
+    key = None
+    repr = "ASFEProtocolResult-"
 
     @pytest.fixture()
     def instance(self, protocol_result):
         return protocol_result
+
+    def test_repr(self, instance):
+        """
+        Overwrites the base `test_repr` call.
+        """
+        assert isinstance(repr(instance), str)
+        assert self.repr in repr(instance)