def test_flat_file_save_with_filename(self): save_dnpy(self.output_path, self.larr0) with open(self.output_path, 'rb') as fp: magic = fp.read(6) self.assertTrue(magic == b'\x93DARRY')
def test_flat_file_save_load_with_file_object(self): save_dnpy(self.output_path, self.larr0) with open(self.output_path, 'rb') as fp: larr1 = load_dnpy(comm=self.comm, file=fp) self.assertTrue(isinstance(larr1, LocalArray)) assert_allclose(self.larr0, larr1)
def test_flat_file_save_load_with_filename(self): save_dnpy(self.output_path, self.larr0) larr1 = load_dnpy(comm=self.comm, file=self.output_path) self.assertTrue(isinstance(larr1, LocalArray)) assert_allclose(self.larr0, larr1)
def _local_save_dnpy_names(local_arr, fnames): from distarray.localapi import save_dnpy fname = fnames[local_arr.comm_rank] save_dnpy(fname, local_arr)
def _local_save_dnpy(local_arr, fname_base): from distarray.localapi import save_dnpy fname = "%s_%s.dnpy" % (fname_base, local_arr.comm_rank) save_dnpy(fname, local_arr)