def test_gw_02(self):
mtrx = csr_matrix(np.identity(6), dtype=int)
r = Utils.delete_rows_csr(mtrx, [0, 1, 2])
e = csr_matrix([[0, 0, 0, 1, 0, 0],
[0, 0, 0, 0, 1, 0],
[0, 0, 0, 0, 0, 1]])
TestUtils.equal_csr_matrix(self, r, e, 'removed row csr_matrix')
def test_gw_01(self):
mtrx = csr_matrix(np.identity(6), dtype=int)
r = Utils.delete_rows_csr(mtrx, [1, 3])
e = csr_matrix([[1, 0, 0, 0, 0, 0],
[0, 0, 1, 0, 0, 0],
[0, 0, 0, 0, 1, 0],
[0, 0, 0, 0, 0, 1]])
TestUtils.equal_csr_matrix(self, r, e, 'removed row csr_matrix')
orig = csr_matrix(np.identity(6), dtype=int)
TestUtils.equal_csr_matrix(self, mtrx, orig, 'original csr_matrix')
def test_bw_01(self):
mtrx = csr_matrix(np.identity(6), dtype=int)
with self.assertRaisesRegex(AssertionError, 'out of bounds'):
# Index out of bounds
Utils.delete_rows_csr(mtrx, [7])
def test_gw_03(self):
mtrx = csr_matrix(np.identity(6), dtype=int)
r = Utils.delete_rows_csr(mtrx, range(6))
e = (0, 6)
TestUtils.equal_tuple_numbers(self, r.shape, e, 'shape of empty matrix')
