def test_hdf5_serialize_unserialize(self):
     filename = None
     try:
         w = world_gen("Dummy", 32, 16, 1, step=Step.get_by_name("full"))
         f = tempfile.NamedTemporaryFile(delete=False)
         f.close()
         filename = f.name
         serialized = save_world_to_hdf5(w, filename)
         unserialized = load_world_to_hdf5(filename)
         self.assertEqual(Set(w.layers.keys()),
                          Set(unserialized.layers.keys()))
         self.assertEqual(w.layers['humidity'].quantiles,
                          unserialized.layers['humidity'].quantiles)
         for l in w.layers.keys():
             self.assertEqual(w.layers[l], unserialized.layers[l],
                              "Comparing %s" % l)
         self.assertTrue(_equal(w.ocean_level, unserialized.ocean_level))
         self.assertEquals(w.seed, unserialized.seed)
         self.assertEquals(w.n_plates, unserialized.n_plates)
         self.assertEquals(w.step, unserialized.step)
         self.assertEqual(sorted(dir(w)), sorted(dir(unserialized)))
         #self.assertEqual(w, unserialized)
     finally:
         if filename:
             os.remove(filename)
Exemple #2
0
 def test_hdf5_serialize_unserialize(self):
     filename = None
     try:
         w = world_gen("Dummy", 32, 16, 1, step=Step.get_by_name("full"))
         f = tempfile.NamedTemporaryFile(delete=False)
         f.close()
         filename = f.name
         serialized = save_world_to_hdf5(w, filename)
         unserialized = load_world_to_hdf5(filename)
         self.assertTrue(
             _equal(w.elevation['data'], unserialized.elevation['data']))
         self.assertEqual(w.elevation['thresholds'],
                          unserialized.elevation['thresholds'])
         self.assertTrue(_equal(w.plates, unserialized.plates))
         self.assertTrue(_equal(w.ocean, unserialized.ocean))
         self.assertTrue(_equal(w.biome, unserialized.biome))
         self.assertTrue(
             _equal(w.humidity['quantiles'],
                    unserialized.humidity['quantiles']))
         self.assertTrue(
             _equal(w.humidity['data'], unserialized.humidity['data']))
         self.assertTrue(_equal(w.humidity, unserialized.humidity))
         self.assertTrue(_equal(w.irrigation, unserialized.irrigation))
         self.assertTrue(_equal(w.permeability, unserialized.permeability))
         self.assertTrue(_equal(w.watermap, unserialized.watermap))
         self.assertTrue(
             _equal(w.precipitation['thresholds'],
                    unserialized.precipitation['thresholds']))
         self.assertTrue(
             _equal(w.precipitation['data'],
                    unserialized.precipitation['data']))
         self.assertTrue(_equal(w.precipitation,
                                unserialized.precipitation))
         self.assertTrue(_equal(w.temperature, unserialized.temperature))
         self.assertTrue(_equal(w.sea_depth, unserialized.sea_depth))
         self.assertTrue(_equal(w.ocean_level, unserialized.ocean_level))
         self.assertTrue(_equal(w.lake_map, unserialized.lake_map))
         self.assertTrue(_equal(w.river_map, unserialized.river_map))
         self.assertTrue(_equal(w.icecap, unserialized.icecap))
         self.assertEquals(w.seed, unserialized.seed)
         self.assertEquals(w.n_plates, unserialized.n_plates)
         self.assertEquals(w.step, unserialized.step)
         self.assertEqual(sorted(dir(w)), sorted(dir(unserialized)))
         #self.assertEqual(w, unserialized)
     finally:
         if filename:
             os.remove(filename)
 def test_hdf5_serialize_unserialize(self):
     filename = None
     try:
         w = world_gen("Dummy", 32, 16, 1, step=Step.get_by_name("full"))
         f = tempfile.NamedTemporaryFile(delete=False)
         f.close()
         filename = f.name
         serialized = save_world_to_hdf5(w, filename)
         unserialized = load_world_to_hdf5(filename)
         self.assertEqual(set(w.layers.keys()), set(unserialized.layers.keys()))
         self.assertEqual(w.layers['humidity'].quantiles, unserialized.layers['humidity'].quantiles)
         for l in w.layers.keys():
             self.assertEqual(w.layers[l], unserialized.layers[l], "Comparing %s" % l)
         self.assertTrue(_equal(w.ocean_level,       unserialized.ocean_level))
         self.assertEquals(w.seed,                   unserialized.seed)
         self.assertEquals(w.n_plates,               unserialized.n_plates)
         self.assertEquals(w.step,                   unserialized.step)
         self.assertEqual(sorted(dir(w)),            sorted(dir(unserialized)))
         #self.assertEqual(w, unserialized)
     finally:
         if filename:
             os.remove(filename)
 def test_hdf5_serialize_unserialize(self):
     filename = None
     try:
         w = world_gen("Dummy", 32, 16, 1, step=Step.get_by_name("full"))
         f = tempfile.NamedTemporaryFile(delete=False)
         f.close()
         filename = f.name
         serialized = save_world_to_hdf5(w, filename)
         unserialized = load_world_to_hdf5(filename)
         self.assertTrue(_equal(w.elevation['data'], unserialized.elevation['data']))
         self.assertEqual(w.elevation['thresholds'], unserialized.elevation['thresholds'])
         self.assertTrue(_equal(w.ocean,             unserialized.ocean))
         self.assertTrue(_equal(w.biome,             unserialized.biome))
         self.assertTrue(_equal(w.humidity['quantiles'], unserialized.humidity['quantiles']))
         self.assertTrue(_equal(w.humidity['data'],  unserialized.humidity['data']))
         self.assertTrue(_equal(w.humidity,          unserialized.humidity))
         self.assertTrue(_equal(w.irrigation,        unserialized.irrigation))
         self.assertTrue(_equal(w.permeability,      unserialized.permeability))
         self.assertTrue(_equal(w.watermap,          unserialized.watermap))
         self.assertTrue(_equal(w.precipitation['thresholds'], unserialized.precipitation['thresholds']))
         self.assertTrue(_equal(w.precipitation['data'], unserialized.precipitation['data']))
         self.assertTrue(_equal(w.precipitation,     unserialized.precipitation))
         self.assertTrue(_equal(w.temperature,       unserialized.temperature))
         self.assertTrue(_equal(w.sea_depth,         unserialized.sea_depth))
         self.assertTrue(_equal(w.ocean_level,       unserialized.ocean_level))
         self.assertTrue(_equal(w.lake_map,          unserialized.lake_map))
         self.assertTrue(_equal(w.river_map,         unserialized.river_map))
         self.assertTrue(_equal(w.icecap,            unserialized.icecap))
         self.assertEquals(w.seed,                   unserialized.seed)
         self.assertEquals(w.n_plates,               unserialized.n_plates)
         self.assertEquals(w.step,                   unserialized.step)
         self.assertEqual(sorted(dir(w)),            sorted(dir(unserialized)))
         #self.assertEqual(w, unserialized)
     finally:
         if filename:
             os.remove(filename)