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)
