Python save_dnpyの例

コード例 #1

ファイル: paralleltest_io.py プロジェクト: 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')

コード例 #2

ファイル: paralleltest_io.py プロジェクト: 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')

コード例 #3

ファイル: paralleltest_io.py プロジェクト: 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)

コード例 #4

ファイル: paralleltest_io.py プロジェクト: 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)

コード例 #5

ファイル: context.py プロジェクト: 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)

コード例 #6

ファイル: context.py プロジェクト: 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)

コード例 #7

ファイル: paralleltest_io.py プロジェクト: 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)

コード例 #8

ファイル: paralleltest_io.py プロジェクト: 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)

コード例 #9

ファイル: context.py プロジェクト: 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)

コード例 #10

ファイル: context.py プロジェクト: 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)