def test_rgba(self): actual = dmp(self.arr3, self.arr_rgba) ind = [0, 1, 5] expected = (self.arr3.take(ind).compressed(), self.arr_rgba.take(ind, axis=0)) assert_array_equal(actual[0], expected[0]) assert_array_equal(actual[1], expected[1])
def test_datetime(self): actual = dmp(self.arr_dt, self.arr3) ind = [0, 1, 5] expected = (self.arr_dt2.take(ind), self.arr3.take(ind).compressed()) self.assert_(np.all(actual[0] == expected[0]) and np.all(actual[1] == expected[1]))
def test_rgba(self): actual = dmp(self.arr3, self.arr_rgba) ind = [0, 1, 5] expected = (self.arr3.take(ind).compressed(), self.arr_rgba.take(ind, axis=0)) self.assert_(np.all(actual[0] == expected[0]) and np.all(actual[1] == expected[1]))
def test_datetime(self): actual = dmp(self.arr_dt, self.arr3) ind = [0, 1, 5] expected = (self.arr_dt2.take(ind), self.arr3.take(ind).compressed()) assert_array_equal(actual[0], expected[0]) assert_array_equal(actual[1], expected[1])
def test_datetime(self): actual = dmp(self.arr_dt, self.arr3) ind = [0, 1, 5] expected = (self.arr_dt2.take(ind), self.arr3.take(ind).compressed()) self.assert_( np.all(actual[0] == expected[0]) and np.all(actual[1] == expected[1]))
def test_rgba(self): actual = dmp(self.arr3, self.arr_rgba) ind = [0, 1, 5] expected = (self.arr3.take(ind).compressed(), self.arr_rgba.take(ind, axis=0)) self.assert_( np.all(actual[0] == expected[0]) and np.all(actual[1] == expected[1]))
def test_string_seq(self): actual = dmp(self.arr_s, self.arr1) ind = [0, 1, 2, 5] expected = (self.arr_s2.take(ind), self.arr2.take(ind))
def test_rgba(self): actual = dmp(self.arr3, self.arr_rgba) ind = [0, 1, 5] expected = (self.arr3[ind].compressed(), self.arr_rgba[ind]) assert_array_equal(actual[0], expected[0]) assert_array_equal(actual[1], expected[1])
def test_string_seq(self): actual = dmp(self.arr_s, self.arr1) ind = [0, 1, 2, 5] expected = (self.arr_s2[ind], self.arr2[ind]) assert_array_equal(actual[0], expected[0]) assert_array_equal(actual[1], expected[1])
def test_bad_first_arg(self): with pytest.raises(ValueError): dmp('a string', self.arr0)
def test_bad_first_arg(self): dmp('a string', self.arr0)
def test_bad_first_arg(self): dmp("a string", self.arr0)