Floating-point comparison during serialization is now a little less s…

…trict.
Mindwerks · Oct 17, 2015 · 008bbbe · 008bbbe
1 parent c84e437
commit 008bbbe
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/serialization_test.py b/tests/serialization_test.py
@@ -36,7 +36,7 @@ def test_pickle_serialize_unserialize(self):
         self.assertEqual(w.sea_depth,               unserialized.sea_depth)
         self.assertEquals(w.seed,                   unserialized.seed)
         self.assertEquals(w.n_plates,               unserialized.n_plates)
-        self.assertEquals(w.ocean_level,            unserialized.ocean_level)
+        self.assertAlmostEqual(w.ocean_level,       unserialized.ocean_level)
         self.assertEquals(w.lake_map,               unserialized.lake_map)
         self.assertEquals(w.river_map,              unserialized.river_map)
         self.assertEquals(w.step,                   unserialized.step)
@@ -60,7 +60,7 @@ def test_protobuf_serialize_unserialize(self):
         self.assertEqual(w.sea_depth,               unserialized.sea_depth)
         self.assertEquals(w.seed,                   unserialized.seed)
         self.assertEquals(w.n_plates,               unserialized.n_plates)
-        self.assertEquals(w.ocean_level,            unserialized.ocean_level)
+        self.assertAlmostEqual(w.ocean_level,       unserialized.ocean_level)
         self.assertEquals(w.lake_map,               unserialized.lake_map)
         self.assertEquals(w.river_map,              unserialized.river_map)
         self.assertEquals(w.step,                   unserialized.step)