def test_unique_3(self): lar = larry(['a', 'b', 'b', 'c', 'a']) u, indices = unique(lar, return_index=True) assert_array_equal(u, np.array(['a', 'b', 'c'], dtype='|S1')) assert_array_equal(indices, np.array([0, 1, 3])) assert_array_equal(lar[indices], np.array(['a', 'b', 'c'], dtype='|S1'))
def test_unique_4(self): lar = larry([1, 2, 6, 4, 2, 3, 2]) u, indices = unique(lar, return_inverse=True) assert_array_equal(u, np.array([1, 2, 3, 4, 6])) assert_array_equal(indices, np.array([0, 1, 4, 3, 1, 2, 1])) assert_array_equal(u[indices], np.array([1, 2, 6, 4, 2, 3, 2]))
def test_unique_3(self): lar = larry(['a', 'b', 'b', 'c', 'a']) u, indices = unique(lar, return_index=True) assert_array_equal(u, np.array(['a', 'b', 'c'])) assert_array_equal(indices, np.array([0, 1, 3])) assert_array_equal(lar[indices], np.array(['a','b','c']))
def test_unique_2(self): lar = larry([[1, 1], [2, 3]]) arr = unique(lar) assert_array_equal(arr, np.array([1, 2, 3]), "la.unique failed")
def test_unique_1(self): arr = unique(larry([1, 1, 2, 2, 3, 3])) assert_array_equal(arr, np.array([1, 2, 3]), "la.unique failed")