def test_redirect_settings_file(self): data = _load_redirect(os.path.join(test_dir, "test_settings.yaml")) self.assertEqual( data, { 'old_module': { 'old_class': { '@class': 'new_class', '@module': 'new_module' } } })
def test_redirect_settings_file(self): data = _load_redirect(os.path.join(test_dir, "test_settings.yaml")) self.assertEqual( data, { "old_module": { "old_class": { "@class": "new_class", "@module": "new_module" } } }, )
# Usage code >>> vor = get_marked_class("VoronoiNN", NNDict) The new class are marked with "_D" """ import json import os from inspect import getfullargspec from typing import List, Dict from monty.json import MontyDecoder, _load_redirect REDIRECT = _load_redirect(os.path.join(os.path.expanduser("~"), ".monty.yaml")) def as_dict(self) -> dict: """ A JSON serializable dict representation of an object. """ d = { "@module": self.__class__.__module__, "@class": self.__class__.__name__, "@version": None } dk = {} spec = getfullargspec(self.__init__) args = spec.args