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