fix bugs in specs test (#184)

instadeepai · Aug 15, 2023 · 873e63d · 873e63d
1 parent 7970dec
commit 873e63d
Showing 1 changed file with 5 additions and 5 deletions.
diff --git 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"][