fix(fw): fix fetch container.kind in eof_test

shemnon · Jun 26, 2024 · b08db3b · b08db3b
1 parent 1b9ebe0
commit b08db3b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/ethereum_test_tools/spec/eof/eof_test.py b/src/ethereum_test_tools/spec/eof/eof_test.py
@@ -165,14 +165,14 @@ def check_container_exception(cls, data: Any) -> Any:
                         )
                     if expect_exception is None:
                         data["expect_exception"] = container.validity_error
-                if "container_kind" in container.model_fields_set:
+                if "kind" in container.model_fields_set:
                     if container_kind is not None:
-                        assert str(container.kind) == container_kind, (
+                        assert container.kind == container_kind, (
                             f"Container kind type {str(container.kind)} "
                             f"does not match test {container_kind}."
                         )
                     if container.kind != ContainerKind.RUNTIME:
-                        data["container_kind"] = str(container.kind)
+                        data["container_kind"] = container.kind
         return data
 
     @classmethod