From aa26d9779f983ff73ba4a9ea16b3739edd624a75 Mon Sep 17 00:00:00 2001 From: Ngo Viet Hoai Bao Date: Mon, 24 Apr 2023 22:30:08 +0700 Subject: [PATCH] fix inplace method testing & other minor tests (#14152) Co-authored-by: CatB1t --- .../test_ivy/helpers/function_testing.py | 33 +++++++------------ 1 file changed, 12 insertions(+), 21 deletions(-) diff --git a/ivy_tests/test_ivy/helpers/function_testing.py b/ivy_tests/test_ivy/helpers/function_testing.py index 5f3f8232e0639..7c5119b6b2c4c 100644 --- 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,