def test_right_rotate_trace(): a = torch.randn(5, 5) for i in range(1, 5): actual = circulant.rotate(a, -i).trace() result = circulant.left_rotate_trace(a, -i) assert(math.fabs(actual - result) < 1e-5)
def test_left_rotate_trace(self): a = torch.randn(5, 5) for i in range(1, 5): actual = circulant.rotate(a, i).trace() result = circulant.left_rotate_trace(a, i) self.assertLess(math.fabs(actual - result), 1e-5)