Python save_dnpy Examples

Example #1

File: paralleltest_io.py Project: jrodgers01d/distarray

    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')

Example #2

File: paralleltest_io.py Project: RaoUmer/distarray

    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')

Example #3

File: paralleltest_io.py Project: jrodgers01d/distarray

 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)

Example #4

File: paralleltest_io.py Project: jrodgers01d/distarray

 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)

Example #5

File: context.py Project: jrodgers01d/distarray

 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)

Example #6

File: context.py Project: jrodgers01d/distarray

 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)

Example #7

File: paralleltest_io.py Project: RaoUmer/distarray

 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)

Example #8

File: paralleltest_io.py Project: RaoUmer/distarray

 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)

Example #9

File: context.py Project: RaoUmer/distarray

 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)

Example #10

File: context.py Project: RaoUmer/distarray

 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)