Python save_hdf5の例

コード例 #1

ファイル: test_hdf5.py プロジェクト: EdwardRaff/chainer

    def test_save(self):
        obj = mock.MagicMock()
        hdf5.save_hdf5(self.temp_file_path, obj, compression=3)

        self.assertEqual(obj.serialize.call_count, 1)
        (serializer,), _ = obj.serialize.call_args
        self.assertIsInstance(serializer, hdf5.HDF5Serializer)
        self.assertEqual(serializer.compression, 3)

コード例 #2

ファイル: test_hdf5.py プロジェクト: smiththth/chainer

    def test_save(self):
        obj = mock.MagicMock()
        hdf5.save_hdf5(self.temp_file_path, obj, compression=3)

        self.assertEqual(obj.serialize.call_count, 1)
        (serializer, ), _ = obj.serialize.call_args
        self.assertIsInstance(serializer, hdf5.HDF5Serializer)
        self.assertEqual(serializer.compression, 3)

コード例 #3

    def setUp(self):
        fd, path = tempfile.mkstemp()
        os.close(fd)
        self.temp_file_path = path

        child = link.Chain(linear=links.Linear(2, 3))
        parent = link.Chain(linear=links.Linear(3, 2), child=child)
        hdf5.save_hdf5(self.temp_file_path, parent)
        self.source = parent

        self.hdf5file = h5py.File(path, 'r')
        self.deserializer = hdf5.HDF5Deserializer(self.hdf5file, strict=False)

コード例 #4

ファイル: test_hdf5.py プロジェクト: delta2323/chainer

    def setUp(self):
        fd, path = tempfile.mkstemp()
        os.close(fd)
        self.temp_file_path = path

        child = link.Chain(linear=links.Linear(2, 3))
        parent = link.Chain(linear=links.Linear(3, 2), child=child)
        hdf5.save_hdf5(self.temp_file_path, parent)
        self.source = parent

        self.hdf5file = h5py.File(path, 'r')
        self.deserializer = hdf5.HDF5Deserializer(self.hdf5file, strict=False)

コード例 #5

ファイル: test_hdf5.py プロジェクト: smiththth/chainer

 def test_save_optimizer2(self):
     hdf5.save_hdf5(self.temp_file_path, self.optimizer)
     with h5py.File(self.temp_file_path) as h5:
         self._check_group(h5, ('Wp', 'epoch', 't'))

コード例 #6

ファイル: test_hdf5.py プロジェクト: smiththth/chainer

 def test_save_chain2(self):
     hdf5.save_hdf5(self.temp_file_path, self.parent)
     with h5py.File(self.temp_file_path) as h5:
         self._check_group(h5, ('Wp', ))

コード例 #7

ファイル: test_hdf5.py プロジェクト: tkng/chainer

 def test_save_optimizer2(self):
     hdf5.save_hdf5(self.temp_file_path, self.optimizer)
     with h5py.File(self.temp_file_path) as h5:
         self._check_group(h5, ('Wp', 'epoch', 't'))

コード例 #8

ファイル: test_hdf5.py プロジェクト: tkng/chainer

 def test_save_chain2(self):
     hdf5.save_hdf5(self.temp_file_path, self.parent)
     with h5py.File(self.temp_file_path) as h5:
         self._check_group(h5, ('Wp',))