TST: _utils.in1d: add tests

Author: lucascolley

tests/test_funcs.py

@@ -1,4 +1,4 @@
-from __future__ import annotations
+from __future__ import annotations  # https://github.com/pylint-dev/pylint/pull/9990
 
 import contextlib
 import typing
@@ -291,7 +291,7 @@ def test_setdiff1d(self):
         x2 = x1
         assert_equal(setdiff1d(x1, x2, xp=xp).dtype, xp.uint32)
 
-    def test_setdiff1d_unique(self):
+    def test_assume_unique(self):
         x1 = xp.asarray([3, 2, 1])
         x2 = xp.asarray([7, 5, 2])
         expected = xp.asarray([3, 1])

tests/test_utils.py

@@ -0,0 +1,17 @@
+from __future__ import annotations  # https://github.com/pylint-dev/pylint/pull/9990
+
+# data-apis/array-api-strict#6
+import array_api_strict as xp  # type: ignore[import-untyped]  # pyright: ignore[reportMissingTypeStubs]
+from numpy.testing import assert_array_equal
+
+from array_api_extra._lib._utils import in1d
+
+
+# some test coverage already provided by TestSetDiff1D
+class TestIn1D:
+    def test_no_invert_assume_unique(self):
+        x1 = xp.asarray([1, 2, 3])
+        x2 = xp.asarray([3, 4, 5])
+        expected = xp.asarray([False, False, True])
+        actual = in1d(x1, x2, xp=xp)
+        assert_array_equal(actual, expected)

tests/test_version.py

@@ -1,4 +1,4 @@
-from __future__ import annotations
+from __future__ import annotations  # https://github.com/pylint-dev/pylint/pull/9990
 
 import importlib.metadata