adding unique_inverse tests

tests/test_unique_inverse_55.py

@@ -0,0 +1,45 @@
+from unique_inverse import unique_inverse
+import awkward as ak
+import ragged
+
+def test_can_take_none():
+    assert unique_inverse(None) is None
+
+def test_can_take_list():
+    assert unique_inverse([1,2,4,3,4,5,6,20]) is None
+
+def test_can_take_empty_arr():
+    arr = ragged.array([])
+    expected_unique_values = ragged.array([])
+    expected_unique_indices = ragged.array([])
+    unique_values, unique_indices = unique_inverse(arr)
+    assert ak.to_list(unique_values) == ak.to_list(expected_unique_values)
+    assert ak.to_list(unique_indices) == ak.to_list(expected_unique_indices)
+
+def test_can_take_simple_array():
+    arr = ragged.array([1, 2, 2, 3, 3, 3, 4, 4, 4, 4])
+    expected_unique_values = ragged.array([1, 2, 3, 4])
+    expected_unique_indices = ragged.array([0, 1, 3, 6])
+    unique_values, unique_indices = unique_inverse(arr)
+    assert ak.to_list(unique_values) == ak.to_list(expected_unique_values)
+    assert ak.to_list(unique_indices) == ak.to_list(expected_unique_indices)
+
+def test_can_take_normal_array():
+    arr = ragged.array([[1, 2, 2], [3], [3, 3], [4, 4, 4], [4]])
+    expected_unique_values = ragged.array([1, 2, 3, 4])
+    expected_unique_indices = ragged.array([0, 1, 3, 6])
+    unique_values, unique_indices = unique_inverse(arr)
+    assert ak.to_list(unique_values) == ak.to_list(expected_unique_values)
+    assert ak.to_list(unique_indices) == ak.to_list(expected_unique_indices)
+
+def test_can_take_awkward():
+    arr = ragged.array([[[1, 2, 2], 3, 3], [3, 4, 4, 4, 4]])
+    assert unique_inverse(arr) is None
+
+def test_can_take_scalar():
+    arr = ragged.array([5])
+    expected_unique_values = ragged.array([5])
+    expected_unique_indices = ragged.array([0])
+    unique_values, unique_indices = unique_inverse(arr)
+    assert ak.to_list(unique_values) == ak.to_list(expected_unique_values)
+    assert ak.to_list(unique_indices) == ak.to_list(expected_unique_indices)

tests/unique_counts.py

@@ -1,7 +1,12 @@
+import awkward as ak
+import ragged
+import numpy as np
+
+
 def unique_counts(arr):
     if not isinstance(arr, ragged.array):
         print("Input is not a ragged array")
-
+        return None
     if len(arr)==1:
         return [(arr[0], 1)]
 

tests/unique_inverse.py

@@ -1,11 +1,15 @@
+import awkward as ak
+import ragged
+import numpy as np
+
 def unique_inverse(arr):
     if not isinstance(arr, ragged.array):
         print("Input is not a ragged array")
-
+        return None
     if len(arr) == 1:
-        return arr, np.array([0])
+        return (arr, np.array([0]))
 
-    arr_list=ak.ravel(arr)
+    arr_list=ak.ravel(arr._impl)
     arr_np = ak.to_numpy(arr_list)
     unique_elements, first_indices = np.unique(arr_np, return_index=True)
 

tests/unique_values.py

@@ -9,7 +9,7 @@ def unique_values(arr):
     if len(arr)==1:
         return arr
 
-    arr_flat=ak.ravel(arr)
+    arr_flat=ak.ravel(arr._impl)
     arr_np = ak.to_numpy(arr_flat)
     unique_np = np.unique(arr_np)
     unique_arr = ak.from_numpy(unique_np)