def test_save_objects(self): objs = [self.obj1, self.obj2] storage.save_objects(self.path, self.extension, objs) for obj, data in zip(objs, [self.d1, self.d2]): # TODO Ideally the below would work without calling convert. path = "%s/%s%s" % (self.path, convert.uuid_to_base32(obj.uuid), self.extension) self.assertTrue(os.path.exists(path)) loaded_data = storage.load(path) self.assertDictEqual(data, loaded_data)
def save_celestials(path, universe): """ Save all Celestial data to files under the given path. """ return storage.save_objects(path, ".celestial.json", universe.celestials.values())
def save_star_systems(path, universe): """ Save all StarSystem data to files under the given path. """ return storage.save_objects(path, ".star_system.json", universe.star_systems.values())