Merge pull request #1005 from OpenFreeEnergy/fix-1004

Fix tests for issue 1004
OpenFreeEnergy · Nov 7, 2024 · ab9a78a · ab9a78a
2 parents be2bb5e + c5adc20
commit ab9a78a
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 7 deletions.
diff --git a/openfe/tests/protocols/test_rfe_tokenization.py b/openfe/tests/protocols/test_rfe_tokenization.py
@@ -49,7 +49,7 @@ def instance(self, protocol):
 
 class TestRelativeHybridTopologyProtocolUnit(GufeTokenizableTestsMixin):
     cls = openmm_rfe.RelativeHybridTopologyProtocolUnit
-    repr = "RelativeHybridTopologyProtocolUnit(benzene to toluene repeat 2 generation 0)"
+    repr = "RelativeHybridTopologyProtocolUnit(benzene to toluene repeat"
     key = None
 
     @pytest.fixture()
@@ -58,3 +58,10 @@ def instance(self, protocol_unit):
 
     def test_key_stable(self):
         pytest.skip()
+
+    def test_repr(self, instance):
+        """
+        Overwrites the base `test_repr` call.
+        """
+        assert isinstance(repr(instance), str)
+        assert self.repr in repr(instance)
diff --git a/openfe/tests/protocols/test_solvation_afe_tokenization.py b/openfe/tests/protocols/test_solvation_afe_tokenization.py
@@ -59,28 +59,36 @@ def instance(self, protocol):
 
 class TestAbsoluteSolvationSolventUnit(GufeTokenizableTestsMixin):
     cls = openmm_afe.AbsoluteSolvationSolventUnit
-    repr = "AbsoluteSolvationSolventUnit(Absolute Solvation, benzene solvent leg: repeat 2 generation 0)"
+    repr = "AbsoluteSolvationSolventUnit(Absolute Solvation, benzene solvent leg"
     key = None
 
     @pytest.fixture()
     def instance(self, solvent_protocol_unit):
         return solvent_protocol_unit
 
-    def test_key_stable(self):
-        pytest.skip()
+    def test_repr(self, instance):
+        """
+        Overwrites the base `test_repr` call.
+        """
+        assert isinstance(repr(instance), str)
+        assert self.repr in repr(instance)
 
 
 class TestAbsoluteSolvationVacuumUnit(GufeTokenizableTestsMixin):
     cls = openmm_afe.AbsoluteSolvationVacuumUnit
-    repr = "AbsoluteSolvationVacuumUnit(Absolute Solvation, benzene vacuum leg: repeat 2 generation 0)"
+    repr = "AbsoluteSolvationVacuumUnit(Absolute Solvation, benzene vacuum leg"
     key = None
 
     @pytest.fixture()
     def instance(self, vacuum_protocol_unit):
         return vacuum_protocol_unit
 
-    def test_key_stable(self):
-        pytest.skip()
+    def test_repr(self, instance):
+        """
+        Overwrites the base `test_repr` call.
+        """
+        assert isinstance(repr(instance), str)
+        assert self.repr in repr(instance)
 
 
 class TestAbsoluteSolvationProtocolResult(GufeTokenizableTestsMixin):