tweaking tests

ivy-llc · Oct 17, 2022 · 22753be · 22753be
1 parent 8e16bd5
commit 22753be
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/ivy_tests/test_array_api b/ivy_tests/test_array_api
diff --git a/ivy_tests/test_ivy/test_functional/test_extensions.py b/ivy_tests/test_ivy/test_functional/test_extensions.py
@@ -698,8 +698,11 @@ def statistical_dtype_values(draw, *, function):
 # implement test for isin
 @handle_cmd_line_args
 @given(
-    dtype_and_x = helpers.dtype_and_values(available_dtypes=helpers.get_dtypes("numeric"), num_arrays=2),
-    num_positional_args=helpers.num_positional_args(fn_name="isin")
+    dtype_and_x=helpers.dtype_and_values(available_dtypes=helpers.get_dtypes("numeric"),
+                                         num_arrays=2),
+    num_positional_args=helpers.num_positional_args(fn_name="isin"),
+    invert=st.booleans(),
+    assume_unique=st.booleans()
 )
 
 def test_isin(
@@ -713,7 +716,6 @@ def test_isin(
     fw,
 ):
     dtype, values = dtype_and_x
-    assume(len(values) == 2)
     helpers.test_function(
         input_dtypes=dtype,
         as_variable_flags=as_variable,
@@ -724,8 +726,8 @@ def test_isin(
         instance_method=instance_method,
         fw=fw,
         fn_name="isin",
-        element=values[0],
-        test_elements=values[1:],
+        element=np.asarray(values[0], dtype=dtype[0]),
+        test_elements=np.asarray(values[1], dtype=dtype[1]),
     )
 
 @handle_cmd_line_args