def test_serialization(self):
mall = MinimumDistanceNNAll(4)
config = serialize(mall)
self.assertDictEqual(
config, {"@module": "megnet.data.local_env", "@class": "MinimumDistanceNNAll", "cutoff": 4}
)
self.assertTrue(serialize(None) is None)
mall2 = deserialize(config)
self.assertTrue(isinstance(mall2, MinimumDistanceNNAll))
self.assertTrue(mall2.cutoff == 4)
def test_serialization(self):
mall = MinimumDistanceNNAll(4)
config = serialize(mall)
self.assertDictEqual(
config, {
'@module': 'megnet.data.local_env',
'@class': 'MinimumDistanceNNAll',
'cutoff': 4
})
self.assertTrue(serialize(None) is None)
mall2 = deserialize(config)
self.assertTrue(isinstance(mall2, MinimumDistanceNNAll))
self.assertTrue(mall2.cutoff == 4)
def as_dict(self) -> Dict:
"""
Serialize to dict
Returns: (dict) dictionary of information
"""
all_dict = super().as_dict()
if "nn_strategy" in all_dict:
nn_strategy = all_dict.pop("nn_strategy")
all_dict.update({"nn_strategy": local_env.serialize(nn_strategy)})
return all_dict
def as_dict(self) -> Dict:
all_dict = super().as_dict()
if 'nn_strategy' in all_dict:
nn_strategy = all_dict.pop('nn_strategy')
all_dict.update({'nn_strategy': local_env.serialize(nn_strategy)})
return all_dict
