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