def test_one_dim_tensor_below_diag(self): t = TensorBase(np.array([1, 2, 3])) tdiag = t.diag(-1) self.assertTrue(syft.equal(tdiag.data, TensorBase(np.array([[0, 0, 0, 0], [1, 0, 0, 0], [0, 2, 0, 0], [0, 0, 3, 0]]))))
def test_two_dim_tensor_below_diag(self): t = TensorBase(np.array([[0, 1], [2, 3]])) tdiag = t.diag(-1) self.assertTrue(syft.equal(tdiag.data, TensorBase(np.array([2]))))