update for mypy --strict, prepare de-inheritance (Loader/Dumper)0.14.0

1 files changed, 38 insertions, 0 deletions

diff --git a/_test/test_yamlobject.py b/_test/test_yamlobject.py
new file mode 100644
index 0000000..a5e06d8
--- /dev/null
+++ b/_test/test_yamlobject.py
@@ -0,0 +1,38 @@
+# coding: utf-8
+
+import pytest   # NOQA
+import ruamel.yaml
+
+from roundtrip import round_trip, dedent, round_trip_load, round_trip_dump  # NOQA
+
+
+class Monster(ruamel.yaml.YAMLObject):
+    yaml_tag = u'!Monster'
+
+    def __init__(self, name, hp, ac, attacks):
+        self.name = name
+        self.hp = hp
+        self.ac = ac
+        self.attacks = attacks
+
+    def __repr__(self):
+        return "%s(name=%r, hp=%r, ac=%r, attacks=%r)" % (
+            self.__class__.__name__, self.name, self.hp, self.ac, self.attacks)
+
+
+def test_monster():
+    data = ruamel.yaml.load(dedent("""\
+    --- !Monster
+    name: Cave spider
+    hp: [2,6]    # 2d6
+    ac: 16
+    attacks: [BITE, HURT]
+    """), Loader=ruamel.yaml.Loader)
+    # normal dump, keys will be sorted
+    assert ruamel.yaml.dump(data) == dedent("""\
+    !Monster
+    ac: 16
+    attacks: [BITE, HURT]
+    hp: [2, 6]
+    name: Cave spider
+    """)