def test_save_and_load(self):
weights1 = self.model.get_weights()
with ScratchDir('.'):
self.model.save_model('test.hdf5')
model2 = GraphModel.from_file('test.hdf5')
weights2 = model2.get_weights()
self.assertTrue(np.allclose(weights1[0], weights2[0]))
def test_save_and_load(self):
weights1 = self.model.get_weights()
with ScratchDir('.'):
self.model.metadata = {"units": "eV"} # This is just a random
self.model.save_model('test.hdf5')
model2 = GraphModel.from_file('test.hdf5')
self.assertEqual(model2.metadata, {"units": "eV"})
weights2 = model2.get_weights()
self.assertTrue(np.allclose(weights1[0], weights2[0]))