def test_class_yaml(): class DummyClass: pass JAML.register(DummyClass) a = JAML.load('!DummyClass {}') assert type(a) == DummyClass with open( resource_filename( 'jina', '/'.join( ('resources', 'executors.requests.BaseExecutor.yml')))) as fp: b = fp.read() print(b) c = JAML.load(b) print(c) args = set_pea_parser().parse_args([]) with BasePea(args): pass from jina.executors.requests import _defaults assert _defaults is not None
def test_class_yaml(): class DummyClass: pass JAML.register(DummyClass) a = JAML.load('!DummyClass {}') assert type(a) == DummyClass
def register_class(cls): """ Register the class for dumping loading. :param cls: Target class. :return: Registered class. """ reg_cls_set = getattr(cls, '_registered_class', set()) if cls.__name__ not in reg_cls_set: reg_cls_set.add(cls.__name__) setattr(cls, '_registered_class', reg_cls_set) from jina.jaml import JAML JAML.register(cls) return cls