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