def testIsCloseExecution(self): data = np.array([1.05, 1.0, 1.01, np.nan]) data2 = np.array([1.04, 1.0, 1.03, np.nan]) x = tensor(data, chunk_size=2) y = tensor(data2, chunk_size=3) z = isclose(x, y, atol=.01) res = self.executor.execute_tensor(z, concat=True)[0] expected = np.isclose(data, data2, atol=.01) np.testing.assert_equal(res, expected) z = isclose(x, y, atol=.01, equal_nan=True) res = self.executor.execute_tensor(z, concat=True)[0] expected = np.isclose(data, data2, atol=.01, equal_nan=True) np.testing.assert_equal(res, expected) # test sparse data = sps.csr_matrix(np.array([0, 1.0, 1.01, np.nan])) data2 = sps.csr_matrix(np.array([0, 1.0, 1.03, np.nan])) x = tensor(data, chunk_size=2) y = tensor(data2, chunk_size=3) z = isclose(x, y, atol=.01) res = self.executor.execute_tensor(z, concat=True)[0] expected = np.isclose(data.toarray(), data2.toarray(), atol=.01) np.testing.assert_equal(res, expected) z = isclose(x, y, atol=.01, equal_nan=True) res = self.executor.execute_tensor(z, concat=True)[0] expected = np.isclose(data.toarray(), data2.toarray(), atol=.01, equal_nan=True) np.testing.assert_equal(res, expected)
def testIsclose(self): t1 = ones((2, 3), dtype='f4', chunk_size=2) atol = 1e-4 rtol = 1e-5 equal_nan = True t = isclose(t1, 2, atol=atol, rtol=rtol, equal_nan=equal_nan) self.assertIsInstance(t.op, TensorIsclose) self.assertEqual(t.op.atol, atol) self.assertEqual(t.op.rtol, rtol) self.assertEqual(t.op.equal_nan, equal_nan) t.tiles() self.assertIsInstance(t.chunks[0].op, TensorIsclose) self.assertEqual(t.chunks[0].op.atol, atol) self.assertEqual(t.chunks[0].op.rtol, rtol) self.assertEqual(t.chunks[0].op.equal_nan, equal_nan) t1 = ones((2, 3), dtype='f4', chunk_size=2) t2 = ones((2, 3), dtype='f4', chunk_size=2) atol = 1e-4 rtol = 1e-5 equal_nan = True t = isclose(t1, t2, atol=atol, rtol=rtol, equal_nan=equal_nan) self.assertIsInstance(t.op, TensorIsclose) self.assertEqual(t.op.atol, atol) self.assertEqual(t.op.rtol, rtol) self.assertEqual(t.op.equal_nan, equal_nan) t.tiles() self.assertIsInstance(t.chunks[0].op, TensorIsclose) self.assertEqual(t.chunks[0].op.atol, atol) self.assertEqual(t.chunks[0].op.rtol, rtol) self.assertEqual(t.chunks[0].op.equal_nan, equal_nan)
def test_isclose(): t1 = ones((2, 3), dtype='f4', chunk_size=2) atol = 1e-4 rtol = 1e-5 equal_nan = True t = isclose(t1, 2, atol=atol, rtol=rtol, equal_nan=equal_nan) assert isinstance(t.op, TensorIsclose) assert t.op.atol == atol assert t.op.rtol == rtol assert t.op.equal_nan == equal_nan t = tile(t) assert isinstance(t.chunks[0].op, TensorIsclose) assert t.chunks[0].op.atol == atol assert t.chunks[0].op.rtol == rtol assert t.chunks[0].op.equal_nan == equal_nan t1 = ones((2, 3), dtype='f4', chunk_size=2) t2 = ones((2, 3), dtype='f4', chunk_size=2) atol = 1e-4 rtol = 1e-5 equal_nan = True t = isclose(t1, t2, atol=atol, rtol=rtol, equal_nan=equal_nan) assert isinstance(t.op, TensorIsclose) assert t.op.atol == atol assert t.op.rtol == rtol assert t.op.equal_nan == equal_nan t = tile(t) assert isinstance(t.chunks[0].op, TensorIsclose) assert t.chunks[0].op.atol == atol assert t.chunks[0].op.rtol == rtol assert t.chunks[0].op.equal_nan == equal_nan
def test_is_close_execution(setup): data = np.array([1.05, 1.0, 1.01, np.nan]) data2 = np.array([1.04, 1.0, 1.03, np.nan]) x = tensor(data, chunk_size=2) y = tensor(data2, chunk_size=3) z = isclose(x, y, atol=.01) res = z.execute().fetch() expected = np.isclose(data, data2, atol=.01) np.testing.assert_equal(res, expected) z = isclose(x, y, atol=.01, equal_nan=True) res = z.execute().fetch() expected = np.isclose(data, data2, atol=.01, equal_nan=True) np.testing.assert_equal(res, expected) # test tensor with scalar z = isclose(x, 1.0, atol=.01) res = z.execute().fetch() expected = np.isclose(data, 1.0, atol=.01) np.testing.assert_equal(res, expected) z = isclose(1.0, y, atol=.01) res = z.execute().fetch() expected = np.isclose(1.0, data2, atol=.01) np.testing.assert_equal(res, expected) z = isclose(1.0, 2.0, atol=.01) res = z.execute().fetch() expected = np.isclose(1.0, 2.0, atol=.01) np.testing.assert_equal(res, expected) # test sparse data = sps.csr_matrix(np.array([0, 1.0, 1.01, np.nan])) data2 = sps.csr_matrix(np.array([0, 1.0, 1.03, np.nan])) x = tensor(data, chunk_size=2) y = tensor(data2, chunk_size=3) z = isclose(x, y, atol=.01) res = z.execute().fetch() expected = np.isclose(data.toarray(), data2.toarray(), atol=.01) np.testing.assert_equal(res, expected) z = isclose(x, y, atol=.01, equal_nan=True) res = z.execute().fetch() expected = np.isclose(data.toarray(), data2.toarray(), atol=.01, equal_nan=True) np.testing.assert_equal(res, expected)