def testFlatnonzeroExecution(self):
x = arange(-2, 3, chunk_size=2)
t = flatnonzero(x)
res = self.executor.execute_tensor(t, concat=True)[0]
expected = np.flatnonzero(np.arange(-2, 3))
np.testing.assert_equal(res, expected)
def test_flatnonzero_execution(setup):
x = arange(-2, 3, chunk_size=2)
t = flatnonzero(x)
res = t.execute().fetch()
expected = np.flatnonzero(np.arange(-2, 3))
np.testing.assert_equal(res, expected)