def test_remove_record_from_local_tracking(self): try: # Remove remove_record_from_local_tracking(self.fake_job_dict.keys()[0], config=self.config) dict_copy = self.fake_job_dict.copy() dict_copy.pop("job_name") # Can I use this in a test? I guess I'm double-testing with get_shelve_database(config=self.config) as db: self.assertEqual(dict_copy, db) finally: with get_shelve_database(config=self.config) as db: db = {} db.update(self.fake_job_dict)
def test_get_shelve_database(self): with get_shelve_database(config=self.config) as db: assert(db)