diff --git a/jumanji/specs_test.py b/jumanji/specs_test.py index f3c94fd29..1777fd163 100644 --- a/jumanji/specs_test.py +++ b/jumanji/specs_test.py @@ -143,16 +143,16 @@ def test_spec__validate(self, triply_nested_spec: specs.Spec) -> None: assert isinstance(triply_nested, TriplyNested) def test_spec__replace(self, triply_nested_spec: specs.Spec) -> None: - arg_list = ["doubly_nested", "bounded_array", "discrete_array"] + arg_list = ["bounded_array", "doubly_nested", "doubly_nested", "discrete_array"] modified_specs = [ triply_nested_spec["bounded_array"].replace(name="wrong_name"), triply_nested_spec["doubly_nested"].replace( - y=triply_nested_spec["doubly_nested"]["discrete_array"].replace( - num_values=2 - ) + discrete_array=triply_nested_spec["doubly_nested"][ + "discrete_array" + ].replace(num_values=2) ), triply_nested_spec["doubly_nested"].replace( - doubly_nested=triply_nested_spec["doubly_nested"][ + singly_nested=triply_nested_spec["doubly_nested"][ "singly_nested" ].replace( bounded_array=triply_nested_spec["doubly_nested"]["singly_nested"][