refactor(substreams-testing): Remove shallow functions, be more 🐍

testing/src/runner/models.py

@@ -50,10 +50,6 @@ class ProtocolComponentExpectation(BaseModel):
 
         return "\n".join(differences)
 
-    @staticmethod
-    def from_dto(dto: ProtocolComponent) -> "ProtocolComponentExpectation":
-        return ProtocolComponentExpectation(**dto.dict())
-
 
 class ProtocolComponentWithTestConfig(ProtocolComponentExpectation):
     """Represents a ProtocolComponent with its main attributes and test configuration."""
@@ -63,9 +59,6 @@ class ProtocolComponentWithTestConfig(ProtocolComponentExpectation):
         description="Flag indicating whether to skip simulation for this component",
     )
 
-    def into_protocol_component(self) -> ProtocolComponentExpectation:
-        return ProtocolComponentExpectation(**self.dict())
-
 
 class IntegrationTest(BaseModel):
     """Configuration for an individual test."""

testing/src/runner/runner.py

@@ -137,9 +137,9 @@ class TestRunner:
                         f"'{comp_id}' not found in protocol components."
                     )
 
-                diff = ProtocolComponentExpectation.from_dto(
-                    components_by_id[comp_id]
-                ).compare(expected_component.into_protocol_component())
+                diff = ProtocolComponentExpectation(
+                    **components_by_id[comp_id].dict()
+                ).compare(ProtocolComponentExpectation(**expected_component.dict()))
                 if diff is not None:
                     return TestResult.Failed(diff)