def test_return_new_float_tensor(self): t1 = TensorBase(np.array([1, 1, 1])) t2 = t1.new(np.array([1., 1., 2.])) self.assertTrue(t2.data.dtype == np.float64)
def test_return_new_int_tensor(self): t1 = TensorBase(np.array([1, 1, 1])) t2 = t1.new(np.array([1, 1, 2])) self.assertTrue(t2.data.dtype == np.int64)
def test_encrypted_error(self): t1 = TensorBase(np.array([1, 1, 1]), encrypted=True) t2 = t1.new([1, 1, 2], encrypted=True) self.assertEqual(t2, NotImplemented)