Exemplo n.º 1
0
    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')
Exemplo n.º 2
0
    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')
Exemplo n.º 3
0
 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)
Exemplo n.º 4
0
 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)
Exemplo n.º 5
0
 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)
Exemplo n.º 6
0
 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)
Exemplo n.º 7
0
 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)
Exemplo n.º 8
0
 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)
Exemplo n.º 9
0
 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)
Exemplo n.º 10
0
 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)