def testFrexpExecution(self): data1 = np.random.random((5, 9, 4)) arr1 = tensor(data1.copy(), chunk_size=3) o1, o2 = frexp(arr1) o = o1 + o2 res = self.executor.execute_tensor(o, concat=True)[0] expected = sum(np.frexp(data1)) self.assertTrue(np.allclose(res, expected)) arr1 = tensor(data1.copy(), chunk_size=3) o1 = zeros(data1.shape, chunk_size=3) o2 = zeros(data1.shape, dtype='i8', chunk_size=3) frexp(arr1, o1, o2) o = o1 + o2 res = self.executor.execute_tensor(o, concat=True)[0] expected = sum(np.frexp(data1)) self.assertTrue(np.allclose(res, expected)) data1 = sps.random(5, 9, density=.1) arr1 = tensor(data1.copy(), chunk_size=3) o1, o2 = frexp(arr1) o = o1 + o2 res = self.executor.execute_tensor(o, concat=True)[0] expected = sum(np.frexp(data1.toarray())) np.testing.assert_equal(res.toarray(), expected)
def testFrexp(self): t1 = ones((3, 4, 5), chunks=2) op_type = type(t1.op) o1, o2 = frexp(t1) self.assertIs(o1.op, o2.op) self.assertNotEqual(o1.dtype, o2.dtype) o1, o2 = frexp(t1, t1) self.assertIs(o1, t1) self.assertIsNot(o1.inputs[0], t1) self.assertIsInstance(o1.inputs[0].op, op_type) self.assertIsNot(o2.inputs[0], t1)
def testFrexp(self): t1 = ones((3, 4, 5), chunk_size=2) t2 = empty((3, 4, 5), dtype=np.float_, chunk_size=2) op_type = type(t1.op) o1, o2 = frexp(t1) self.assertIs(o1.op, o2.op) self.assertNotEqual(o1.dtype, o2.dtype) o1, o2 = frexp(t1, t1) self.assertIs(o1, t1) self.assertIsNot(o1.inputs[0], t1) self.assertIsInstance(o1.inputs[0].op, op_type) self.assertIsNot(o2.inputs[0], t1) o1, o2 = frexp(t1, t2, where=t1 > 0) op_type = type(t2.op) self.assertIs(o1, t2) self.assertIsNot(o1.inputs[0], t1) self.assertIsInstance(o1.inputs[0].op, op_type) self.assertIsNot(o2.inputs[0], t1)