def test_new_pantry(self): p = pantry.open(self.filename) self.assertEqual(p.db, {}) # should be empty p.db['Test'] = True p.close() q = pantry.open(self.filename) self.assertTrue(q.db['Test'])
def test_pantry_custom_class(self): p = pantry.open(self.filename) custom = CustomPantry(True, False, a=1, b=2) p.db = custom self.assertEqual(p.db.kwargs['a'], 1) p.close() q = pantry.open(self.filename) self.assertEqual(q.db.kwargs['a'], 1)
def test_pantry_custom_class_defs(self): p = pantry.open(self.filename) p.db = CustomPantry(1, 2, 3, 4, 5) self.assertEqual(sum(p.db.args), sum((1, 2, 3, 4, 5))) p.close() with pantry(self.filename) as q: self.assertEqual(q.do_math(), sum((1, 2, 3, 4, 5)))
def test_immediate_save(self): db = {'a': 1, 'b': 2} pantry.store(self.filename, db) p = pantry.open(self.filename) self.assertEqual(p.db['a'], 1) p.db['b'] = 3 p.close() with pantry(self.filename) as db: self.assertEqual(db['b'], 3)
def test_no_file_pantry(self): p = pantry.open(self.filename+'new') self.assertEqual(p.db, {}) # should be empty
def test_no_file_pantry(self): p = pantry.open(self.filename + 'new') self.assertEqual(p.db, {}) # should be empty