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

2024-08-08 00:12:34 +02:00
parent 1f9fe8d583
commit 139f7ac3f5
2 changed files with 3 additions and 10 deletions
--- 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)