diff options
Diffstat (limited to '_test/test_version.py')
-rw-r--r-- | _test/test_version.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/_test/test_version.py b/_test/test_version.py index 742ec36..e110eed 100644 --- a/_test/test_version.py +++ b/_test/test_version.py @@ -8,7 +8,9 @@ from roundtrip import dedent, round_trip, round_trip_load def load(s, version=None): import ruamel.yaml # NOQA - return ruamel.yaml.round_trip_load(dedent(s), version) + yaml = ruamel.yaml.YAML() + yaml.version = version + return yaml.load(dedent(s)) class TestVersions: |