def test_set_bug_2(self):
mat = [[0, 0, 0, 1], [0, 0, 0, 1], [0, 0, 0, 1], [1, 0, 0, 0]]
n = 4
csr = CSR()
csr.load(mat)
for i in range(n):
for j in range(n):
self.assertEquals(mat[i][j], csr.get(i, j))
rand = random.random()
print(mat)
print(csr)
mat[1][1] = rand
csr.set(1, 1, rand)
print(mat)
print(csr)
self.assertEquals(mat[1][0], csr.get(1, 0))
print(mat)
print(csr)
for i in range(n):
for j in range(n):
self.assertEquals(mat[i][j], csr.get(i, j))
def test_set_bugfromadd(self):
mat = [[1, 1, 0], [1, 0, 0], [0, 1, 0]]
csr1 = CSR(mat)
print(mat)
print(csr1)
csr1.set(0, 2, 3)
mat[0][2] = 3
print(mat)
print(csr1)
for i in range(3):
for j in range(3):
print(i, j, mat[i][j], csr1.get(i, j))
self.assertEquals(mat[i][j], csr1.get(i, j))
def test_set_Diag(self):
for n in [4, 10, 50, 100]:
mat2 = randomizeMatrix(n)
csr = CSR()
csr.load(mat2)
for i in range(n):
rand = random.random()
mat2[i][i] = rand
csr.set(i, i, rand)
print(mat2)
self.assertEquals(rand, csr.get(i, i))
csr2 = CSR(mat2)
for i in range(n):
for j in range(n):
self.assertEquals(mat2[i][j], csr2.get(i, j))
def test_get_simple(self):
mat = [[1, 0, 0, 0], [1, 0, 0, 0], [1, 0, 0, 0], [1, 0, 0, 0]]
n = 4
csr = CSR(mat)
for i in range(n):
for j in range(n):
print(i, j)
self.assertEquals(mat[i][j], csr.get(i, j))
def test_get(self):
for n in [10, 50, 100, 500]:
with self.subTest(i=n):
mat2 = randomizeMatrix(n)
csr_sci = sparse.csr_matrix(mat2)
csr = CSR()
csr.load(mat2)
for i in range(n):
for j in range(n):
self.assertEquals(mat2[i][j], csr.get(i, j))
def test_set_bug_diag(self):
mat = [[0, 0, 1], [1, 0, 0], [0, 1, 0]]
n = len(mat)
csr = CSR(mat)
print(mat)
print(csr)
i, j = 1, 1
rand = random.random()
mat[i][j] = rand
csr.set(i, j, rand)
print(mat)
print(csr)
self.assertEquals(mat[i][j], csr.get(i, j))
def test_set(self):
for n in [4, 10, 50, 100]:
mat2 = randomizeMatrix(n)
csr = CSR(mat2)
for i in range(n):
for j in range(n):
rand = random.random()
mat2[i][j] = rand
csr.set(i, j, rand)
# self.assertEquals(rand, csr.get(i,j))
csr2 = CSR(mat2)
for i in range(n):
for j in range(n):
self.assertEquals(mat2[i][j], csr2.get(i, j))