diff options
Diffstat (limited to '_test/test_class_register.py')
-rw-r--r-- | _test/test_class_register.py | 47 |
1 files changed, 25 insertions, 22 deletions
diff --git a/_test/test_class_register.py b/_test/test_class_register.py index d996269..fdd0275 100644 --- a/_test/test_class_register.py +++ b/_test/test_class_register.py @@ -4,33 +4,36 @@ testing of YAML.register_class and @yaml_object """ +from typing import Any +from ruamel.yaml.comments import TaggedScalar, CommentedMap # NOQA + from roundtrip import YAML -class User0(object): - def __init__(self, name, age): +class User0: + def __init__(self, name: str, age: int) -> None: self.name = name self.age = age -class User1(object): +class User1: yaml_tag = '!user' - def __init__(self, name, age): + def __init__(self, name: str, age: int) -> None: self.name = name self.age = age @classmethod - def to_yaml(cls, representer, node): + def to_yaml(cls, representer: Any, node: Any) -> Any: return representer.represent_scalar(cls.yaml_tag, '{.name}-{.age}'.format(node, node)) @classmethod - def from_yaml(cls, constructor, node): + def from_yaml(cls, constructor: Any, node: Any) -> Any: return cls(*node.value.split('-')) -class TestRegisterClass(object): - def test_register_0_rt(self): +class TestRegisterClass: + def test_register_0_rt(self) -> None: yaml = YAML() yaml.register_class(User0) ys = """ @@ -41,7 +44,7 @@ class TestRegisterClass(object): d = yaml.load(ys) yaml.dump(d, compare=ys, unordered_lines=True) - def test_register_0_safe(self): + def test_register_0_safe(self) -> None: # default_flow_style = None yaml = YAML(typ='safe') yaml.register_class(User0) @@ -51,7 +54,7 @@ class TestRegisterClass(object): d = yaml.load(ys) yaml.dump(d, compare=ys) - def test_register_0_unsafe(self): + def test_register_0_unsafe(self) -> None: # default_flow_style = None yaml = YAML(typ='unsafe') yaml.register_class(User0) @@ -61,7 +64,7 @@ class TestRegisterClass(object): d = yaml.load(ys) yaml.dump(d, compare=ys) - def test_register_1_rt(self): + def test_register_1_rt(self) -> None: yaml = YAML() yaml.register_class(User1) ys = """ @@ -70,7 +73,7 @@ class TestRegisterClass(object): d = yaml.load(ys) yaml.dump(d, compare=ys) - def test_register_1_safe(self): + def test_register_1_safe(self) -> None: yaml = YAML(typ='safe') yaml.register_class(User1) ys = """ @@ -79,7 +82,7 @@ class TestRegisterClass(object): d = yaml.load(ys) yaml.dump(d, compare=ys) - def test_register_1_unsafe(self): + def test_register_1_unsafe(self) -> None: yaml = YAML(typ='unsafe') yaml.register_class(User1) ys = """ @@ -89,15 +92,15 @@ class TestRegisterClass(object): yaml.dump(d, compare=ys) -class TestDecorator(object): - def test_decorator_implicit(self): +class TestDecorator: + def test_decorator_implicit(self) -> None: from ruamel.yaml import yaml_object yml = YAML() @yaml_object(yml) - class User2(object): - def __init__(self, name, age): + class User2: + def __init__(self, name: str, age: int) -> None: self.name = name self.age = age @@ -109,27 +112,27 @@ class TestDecorator(object): d = yml.load(ys) yml.dump(d, compare=ys, unordered_lines=True) - def test_decorator_explicit(self): + def test_decorator_explicit(self) -> None: from ruamel.yaml import yaml_object yml = YAML() @yaml_object(yml) - class User3(object): + class User3: yaml_tag = '!USER' - def __init__(self, name, age): + def __init__(self, name: str, age: int) -> None: self.name = name self.age = age @classmethod - def to_yaml(cls, representer, node): + def to_yaml(cls, representer: Any, node: Any) -> Any: return representer.represent_scalar( cls.yaml_tag, '{.name}-{.age}'.format(node, node) ) @classmethod - def from_yaml(cls, constructor, node): + def from_yaml(cls, constructor: Any, node: Any) -> Any: return cls(*node.value.split('-')) ys = """ |