def test_contains(self): """Make sure we can check if things are present""" x = PySparseMatFloat(2,3) self.assertEqual(x.contains(1,2), 0) x.insert(1,2,20) self.assertEqual(x.contains(1,2), 1) x.insert(1,2,10) self.assertEqual(x.contains(1,2), 1) x.erase(1,2) self.assertEqual(x.contains(1,2), 0) x.insert(1,2,0.0) self.assertEqual(x.contains(1,2), 0)
def test_erase(self): """make sure we can erase""" x = PySparseMatFloat(2,3) x.insert(1,2,10) self.assertEqual(x.get(1,2), 10.0) x.erase(1,2) self.assertEqual(x.get(1,2), 0.0) self.assertEqual(x.contains(1,2), 0)