fix inplace method testing & other minor tests (#14152)

Co-authored-by: CatB1t <[email protected]>
ivy-llc · Apr 24, 2023 · aa26d97 · aa26d97
1 parent 0fbaaa4
commit aa26d97
Showing 1 changed file with 12 additions and 21 deletions.
diff --git a/ivy_tests/test_ivy/helpers/function_testing.py b/ivy_tests/test_ivy/helpers/function_testing.py
@@ -5,6 +5,7 @@
 import types
 import importlib
 import inspect
+from collections import OrderedDict
 
 try:
     import jsonpickle
@@ -1193,8 +1194,8 @@ def test_method(
         for v, d in zip(init_flags.as_variable, init_input_dtypes)
     ]
 
-    # Create Args
-    args_constructor, kwargs_constructor, *_ = create_args_kwargs(
+    # Save original constructor data for inplace operations
+    con_data = OrderedDict(
         args_np=args_np_constructor,
         arg_np_vals=con_arg_np_vals,
         args_idxs=con_args_idxs,
@@ -1205,6 +1206,10 @@ def test_method(
         test_flags=init_flags,
         on_device=on_device,
     )
+    org_con_data = copy.deepcopy(con_data)
+
+    # Create Args
+    args_constructor, kwargs_constructor = create_args_kwargs(**con_data)
     # end constructor #
 
     # method arguments #
@@ -1290,14 +1295,10 @@ def test_method(
         process = ground_truth_backend[1]
         try:
             process.stdin.write("3" + "\n")
-            process.stdin.write(jsonpickle.dumps(args_np_constructor) + "\n")
-            process.stdin.write(jsonpickle.dumps(con_arg_np_vals) + "\n")
-            process.stdin.write(jsonpickle.dumps(con_args_idxs) + "\n")
-            process.stdin.write(jsonpickle.dumps(kwargs_np_constructor) + "\n")
-            process.stdin.write(jsonpickle.dumps(con_kwarg_np_vals) + "\n")
-            process.stdin.write(jsonpickle.dumps(con_kwargs_idxs) + "\n")
-            process.stdin.write(jsonpickle.dumps(init_input_dtypes) + "\n")
-            process.stdin.write(jsonpickle.dumps(init_flags) + "\n")
+            # constructor
+            for arg in list(org_con_data.values())[:-1]:
+                process.stdin.write(jsonpickle.dumps(arg) + "\n")
+            # method
             process.stdin.write(jsonpickle.dumps(args_np_method) + "\n")
             process.stdin.write(jsonpickle.dumps(met_arg_np_vals) + "\n")
             process.stdin.write(jsonpickle.dumps(met_args_idxs) + "\n")
@@ -1332,17 +1333,7 @@ def test_method(
     else:
         ivy.set_backend(ground_truth_backend)
         ivy.set_default_device(on_device)
-        args_gt_constructor, kwargs_gt_constructor = create_args_kwargs(
-            args_np=args_np_constructor,
-            arg_np_vals=con_arg_np_vals,
-            args_idxs=con_args_idxs,
-            kwargs_np=kwargs_np_constructor,
-            kwarg_np_vals=con_kwarg_np_vals,
-            kwargs_idxs=con_kwargs_idxs,
-            input_dtypes=init_input_dtypes,
-            test_flags=init_flags,
-            on_device=on_device,
-        )
+        args_gt_constructor, kwargs_gt_constructor = create_args_kwargs(**org_con_data)
         args_gt_method, kwargs_gt_method = create_args_kwargs(
             args_np=args_np_method,
             arg_np_vals=met_arg_np_vals,