def test_pickle(self):
rep = SphericalInvariants(**self.hypers)
cosine_kernel = Kernel(rep,
name="Cosine",
target_type="Structure",
zeta=2)
serialized = pickle.dumps(cosine_kernel)
cosine_kernel_ = pickle.loads(serialized)
self.assertTrue(to_dict(cosine_kernel) == to_dict(cosine_kernel_))
def test_serialization(self):
rep = SphericalExpansion(**self.hypers)
rep_dict = to_dict(rep)
rep_copy = from_dict(rep_dict)
rep_copy_dict = to_dict(rep_copy)
self.assertTrue(rep_dict == rep_copy_dict)
def test_serialization(self):
rep = SortedCoulombMatrix(**self.hypers)
rep_dict = to_dict(rep)
rep_copy = from_dict(rep_dict)
rep_copy_dict = to_dict(rep_copy)
self.assertTrue(rep_dict == rep_copy_dict)
def test_serialization(self):
rep = SphericalInvariants(**self.hypers)
for target_type in ["Atom", "Structure"]:
cosine_kernel = Kernel(rep, name="Cosine", target_type=target_type, zeta=2)
cosine_kernel_dict = to_dict(cosine_kernel)
cosine_kernel_copy = from_dict(cosine_kernel_dict)
cosine_kernel_copy_dict = to_dict(cosine_kernel_copy)
self.assertTrue(cosine_kernel_dict == cosine_kernel_copy_dict)
def test_pickle(self):
rep = SortedCoulombMatrix(**self.hypers)
serialized = pickle.dumps(rep)
rep_ = pickle.loads(serialized)
self.assertTrue(to_dict(rep) == to_dict(rep_))
def test_pickle(self):
rep = SphericalInvariants(**self.hypers)
serialized = pickle.dumps(rep)
rep_ = pickle.loads(serialized)
self.assertTrue(to_dict(rep) == to_dict(rep_))
