def test_fmod_number(self):
t1 = TensorBase(np.array([-3, -2, -1, 1, 2, 3]))
t1.fmod_(2)
self.assertTrue(np.array_equal(t1.data, np.array([-1, 0, -1, 1, 0, 1])))
t2 = TensorBase(np.array([-3.5, -2.5, -1.5, 1.5, 2.5, 3.5]))
t2.fmod_(2.)
self.assertTrue(np.array_equal(t2.data, np.array([-1.5, -0.5, -1.5, 1.5, 0.5, 1.5])))
def test_fmod_tensor(self):
t1 = TensorBase(np.array([-3, -2, -1, 1, 2, 3]))
divisor = np.array([2] * 6)
t1.fmod_(divisor)
self.assertTrue(np.array_equal(t1.data, np.array([-1, 0, -1, 1, 0, 1])))
t2 = TensorBase(np.array([-3.5, -2.5, -1.5, 1.5, 2.5, 3.5]))
divisor = np.array([2.] * 6)
t2.fmod_(divisor)
self.assertTrue(np.array_equal(t2.data, np.array([-1.5, -0.5, -1.5, 1.5, 0.5, 1.5])))
def test_fmod_(self):
t = TensorBase([[-2, -3], [4, 1]])
t.fmod_(2)
self.assertTrue(np.array_equal(t.data, np.array([[0, -1], [0, 1]])))
