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]])))