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