def test_DDM_setitem(): dm = DDM.zeros((3, 3), ZZ) dm.setitem(0, 0, 1) dm.setitem(1, -2, 1) dm.setitem(-1, -1, 1) assert dm == DDM.eye(3, ZZ) raises(IndexError, lambda: dm.setitem(3, 3, 0))
def test_DDM_eye(): ddmz = DDM.eye(3, QQ) f = lambda i, j: QQ(1) if i == j else QQ(0) assert list(ddmz) == [[f(i, j) for i in range(3)] for j in range(3)] assert ddmz.shape == (3, 3) assert ddmz.domain == QQ
def test_DomainMatrix_eye(): A = DomainMatrix.eye(3, QQ) assert A.rep == DDM.eye(3, QQ) assert A.shape == (3, 3) assert A.domain == QQ