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)
