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