def test_set_get(self): d1 = MongoDict(DB_NAME, 'table1') d2 = MongoDict(DB_NAME, 'table2') d3 = MongoDict(DB_NAME, 'table3') d1[1] = 1 d2[2] = 2 d3[3] = 3 self.assertEqual(list(d1.keys()), [1]) self.assertEqual(list(d2.keys()), [2]) self.assertEqual(list(d3.keys()), [3]) with self.assertRaises(KeyError): a = d1[4]
def test_del(self): d = MongoDict('test') d.clear() for i in range(5): d[i] = i del d[0] del d[1] del d[2] self.assertEqual(list(d.keys()), list(range(3, 5))) self.assertEqual(list(d.values()), list(range(3, 5))) with self.assertRaises(KeyError): del d[0]