def testArgwhereExecution(self):
x = arange(6, chunk_size=2).reshape(2, 3)
t = argwhere(x > 1)
res = self.executor.execute_tensor(t, concat=True)[0]
expected = np.argwhere(np.arange(6).reshape(2, 3) > 1)
self.assertTrue(np.array_equal(res, expected))
def test_argwhere():
cond = tensor([[True, False], [False, True]], chunk_size=1)
indices = argwhere(cond)
assert np.isnan(indices.shape[0])
assert indices.shape[1] == 2
indices = tile(indices)
assert indices.nsplits[1] == (1, 1)
def testArgwhere(self):
cond = tensor([[True, False], [False, True]], chunk_size=1)
indices = argwhere(cond)
self.assertTrue(np.isnan(indices.shape[0]))
self.assertEqual(indices.shape[1], 2)
indices = indices.tiles()
self.assertEqual(indices.nsplits[1], (1, 1))
def testArgwhereExecution(self):
x = arange(6, chunk_size=2).reshape(2, 3)
t = argwhere(x > 1)
res = self.executor.execute_tensor(t, concat=True)[0]
expected = np.argwhere(np.arange(6).reshape(2, 3) > 1)
np.testing.assert_array_equal(res, expected)
data = np.asfortranarray(np.random.rand(10, 20))
x = tensor(data, chunk_size=10)
t = argwhere(x > 0.5)
res = self.executor.execute_tensor(t, concat=True)[0]
expected = np.argwhere(data > 0.5)
np.testing.assert_array_equal(res, expected)
self.assertTrue(res.flags['F_CONTIGUOUS'])
self.assertFalse(res.flags['C_CONTIGUOUS'])
def test_argwhere_order():
data = np.asfortranarray([[True, False], [False, True]])
cond = tensor(data, chunk_size=1)
indices = argwhere(cond)
assert indices.flags['F_CONTIGUOUS'] is True
assert indices.flags['C_CONTIGUOUS'] is False
indices = tile(indices)
assert indices.chunks[0].order.value == 'F'
def testArgwhereOrder(self):
data = np.asfortranarray([[True, False], [False, True]])
cond = tensor(data, chunk_size=1)
indices = argwhere(cond)
self.assertTrue(indices.flags['F_CONTIGUOUS'])
self.assertFalse(indices.flags['C_CONTIGUOUS'])
indices = indices.tiles()
self.assertEqual(indices.chunks[0].order.value, 'F')
