diff --git a/testing/src/runner/models.py b/testing/src/runner/models.py index aa79559..992efa5 100644 --- a/testing/src/runner/models.py +++ b/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.""" diff --git a/testing/src/runner/runner.py b/testing/src/runner/runner.py index c25292d..bf17ef3 100644 --- a/testing/src/runner/runner.py +++ b/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)