def test_contains(self):
"""Make sure we can check things exist"""
sm1 = SparseMat(3,4)
for r in range(3):
for c in range(4):
assert (r,c) not in sm1
sm1[1,2] = 0
assert (1,2) not in sm1
sm1[1,2] = 10
assert (1,2) in sm1
sm1.erase(1,2)
assert (1,2) not in sm1
def test_erase(self):
"""Make sure we can get rid of elements"""
sm1 = SparseMat(4,6)
self.assertEqual(sm1[2,3], 0.0)
sm1.erase(2,3)
self.assertEqual(sm1[2,3], 0.0)
sm1[2,3] = 10
self.assertEqual(sm1[2,3], 10.0)
self.assertEqual(sm1._index_rows, [set([]), set([]), set([(2,3)]), set([])])
self.assertEqual(sm1._index_cols, [set([]), set([]), set([]), set([(2,3)]), set([]), set([])])
sm1.erase(2,3)
self.assertEqual(sm1._index_rows, [set([]), set([]), set([]), set([])])
self.assertEqual(sm1._index_cols, [set([]), set([]), set([]), set([]), set([]), set([])])
self.assertEqual(sm1[2,3], 0.0)
self.assertEqual(sm1._index_rows, [set([]), set([]), set([]), set([])])
self.assertEqual(sm1._index_cols, [set([]), set([]), set([]), set([]), set([]), set([])])
