def test_save_load_with_filenames(self): dac = Context(self.client) da = dac.empty((100,), dist={0: 'b'}) output_paths = [temp_filepath() for target in dac.targets] try: dac.save(output_paths, da) db = dac.load(output_paths) self.assertTrue(isinstance(db, DistArray)) self.assertEqual(da, db) finally: for filepath in output_paths: if os.path.exists(filepath): os.remove(filepath)
def test_save_load_with_prefix(self): dac = Context(self.client) da = dac.empty((100,), dist={0: 'b'}) output_path = temp_filepath() try: dac.save(output_path, da) db = dac.load(output_path) self.assertTrue(isinstance(db, DistArray)) self.assertEqual(da, db) finally: for rank in dac.targets: filepath = output_path + "_" + str(rank) + ".dnpy" if os.path.exists(filepath): os.remove(filepath)