def test21(self): test_results_path = self.get_path_to_results() output_file = os.path.join(test_results_path, "test21" + self.store_version() + ".hdf5") if os.path.exists(output_file): os.remove(output_file) stars = Particles(2) stars.x = 1.0 | units.km stars.md = [[[1, 3], [2, 4], [3, 5]], [[4, 6], [5, 7], [6, 8]]] io.write_set_to_file(stars, output_file, "hdf5", version=self.store_version()) loaded = io.read_set_from_file(output_file, "hdf5", close_file=True, version=self.store_version(), copy_history=False) self.assertEqual(loaded[0].md, [[1, 3], [2, 4], [3, 5]]) self.assertEqual(loaded[1].md, [[4, 6], [5, 7], [6, 8]]) previous = loaded.previous_state() self.assertEqual(previous, None)
def test15(self): test_results_path = self.get_path_to_results() output_file = os.path.join(test_results_path, "test15" + self.store_version() + ".hdf5") if os.path.exists(output_file): os.remove(output_file) stars = Particles(2) stars.x = 1.0 | units.km stars.md = [[[1, 3], [2, 4], [3, 5]], [[4, 6], [5, 7], [6, 8]]] io.write_set_to_file(stars, output_file, "hdf5", version=self.store_version()) processor = self.store_factory()(output_file, True, open_for_writing=True) loaded = processor.load() self.get_version_in_store(loaded)[0].md = [[3, 1], [3, 4], [5, 2]] self.assertEqual( self.get_version_in_store(loaded)[0].md, [[3, 1], [3, 4], [5, 2]]) processor.close() loaded = io.read_set_from_file(output_file, "hdf5", version=self.store_version()) self.assertEqual(loaded[0].md, [[3, 1], [3, 4], [5, 2]]) self.assertEqual(loaded[1].md, [[4, 6], [5, 7], [6, 8]])
def test13(self): test_results_path = self.get_path_to_results() output_file = os.path.join(test_results_path, "test13" + self.store_version() + ".hdf5") if os.path.exists(output_file): os.remove(output_file) stars = Particles(2) stars.x = 1.0 | units.km stars.md = [[1, 2, 3], [4, 5, 6]] | units.km io.write_set_to_file(stars, output_file, "hdf5", version=self.store_version()) loaded = io.read_set_from_file(output_file, "hdf5", version=self.store_version()) self.assertEqual(loaded[0].md, [1, 2, 3] | units.km) self.assertEqual(loaded[1].md, [4, 5, 6] | units.km) self.assertEqual(loaded.md[0], [1, 2, 3] | units.km) self.assertEqual(loaded.md[1], [4, 5, 6] | units.km) self.assertEqual( self.get_version_in_store(loaded)[0].md, [1, 2, 3] | units.km)
def test14(self): test_results_path = self.get_path_to_results() output_file = os.path.join(test_results_path, "test14"+self.store_version()+".hdf5") if os.path.exists(output_file): os.remove(output_file) stars = Particles(2) stars.x = 1.0 | units.km stars.md = [[[1,3],[2,4],[3,5]],[[4,6],[5,7],[6,8]]] io.write_set_to_file(stars, output_file, "hdf5", version = self.store_version()) loaded = io.read_set_from_file(output_file, "hdf5", version = self.store_version()) self.assertEquals(loaded[0].md, [[1,3],[2,4],[3,5]]) self.assertEquals(loaded[1].md, [[4,6],[5,7],[6,8]]) self.assertEquals(loaded.md[0], [[1,3],[2,4],[3,5]]) self.assertEquals(loaded.md[1], [[4,6],[5,7],[6,8]])
def test13(self): test_results_path = self.get_path_to_results() output_file = os.path.join(test_results_path, "test13"+self.store_version()+".hdf5") if os.path.exists(output_file): os.remove(output_file) stars = Particles(2) stars.x = 1.0 | units.km stars.md = [[1,2,3],[4,5,6]] | units.km io.write_set_to_file(stars, output_file, "hdf5", version = self.store_version()) loaded = io.read_set_from_file(output_file, "hdf5", version = self.store_version()) self.assertEquals(loaded[0].md, [1,2,3] | units.km) self.assertEquals(loaded[1].md, [4,5,6] | units.km) self.assertEquals(loaded.md[0], [1,2,3] | units.km) self.assertEquals(loaded.md[1], [4,5,6] | units.km) self.assertEquals(self.get_version_in_store(loaded)[0].md, [1,2,3] | units.km)
def test15(self): test_results_path = self.get_path_to_results() output_file = os.path.join(test_results_path, "test15"+self.store_version()+".hdf5") if os.path.exists(output_file): os.remove(output_file) stars = Particles(2) stars.x = 1.0 | units.km stars.md = [[[1,3],[2,4],[3,5]],[[4,6],[5,7],[6,8]]] io.write_set_to_file(stars, output_file, "hdf5", version = self.store_version()) processor = self.store_factory()(output_file, True, open_for_writing = True) loaded = processor.load() self.get_version_in_store(loaded)[0].md = [[3,1],[3,4],[5,2]] self.assertEquals(self.get_version_in_store(loaded)[0].md, [[3,1],[3,4],[5,2]] ) processor.close() loaded = io.read_set_from_file(output_file, "hdf5", version = self.store_version()) self.assertEquals(loaded[0].md, [[3,1],[3,4],[5,2]]) self.assertEquals(loaded[1].md, [[4,6],[5,7],[6,8]])