1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108
|
# coding: utf-8
import pytest # type: ignore # NOQA
from roundtrip import ( # type: ignore
round_trip,
round_trip_dump_all,
round_trip_load_all,
)
class TestDocument:
def test_single_doc_begin_end(self) -> None:
inp = """\
---
- a
- b
...
"""
round_trip(inp, explicit_start=True, explicit_end=True)
def test_multi_doc_begin_end(self) -> None:
inp = """\
---
- a
...
---
- b
...
"""
docs = list(round_trip_load_all(inp))
assert docs == [['a'], ['b']]
out = round_trip_dump_all(docs, explicit_start=True, explicit_end=True)
assert out == '---\n- a\n...\n---\n- b\n...\n'
def test_multi_doc_no_start(self) -> None:
inp = """\
- a
...
---
- b
...
"""
docs = list(round_trip_load_all(inp))
assert docs == [['a'], ['b']]
def test_multi_doc_no_end(self) -> None:
inp = """\
- a
---
- b
"""
docs = list(round_trip_load_all(inp))
assert docs == [['a'], ['b']]
def test_multi_doc_ends_only(self) -> None:
# this is ok in 1.2
inp = """\
- a
...
- b
...
"""
docs = list(round_trip_load_all(inp, version=(1, 2)))
assert docs == [['a'], ['b']]
def test_single_scalar_comment(self) -> None:
import ruyaml as yaml
inp = """\
one # comment
two
"""
with pytest.raises(yaml.parser.ParserError):
d = list(round_trip_load_all(inp, version=(1, 2))) # NOQA
def test_scalar_after_seq_document(self) -> None:
import ruyaml as yaml
inp = """\
[ 42 ]
hello
"""
with pytest.raises(yaml.parser.ParserError):
d = list(round_trip_load_all(inp, version=(1, 2))) # NOQA
def test_yunk_after_explicit_document_end(self) -> None:
import ruyaml as yaml
inp = """\
hello: world
... this is no comment
"""
with pytest.raises(yaml.parser.ParserError):
d = list(round_trip_load_all(inp, version=(1, 2))) # NOQA
def test_multi_doc_ends_only_1_1(self) -> None:
import ruyaml as yaml
# this is not ok in 1.1
with pytest.raises(yaml.parser.ParserError):
inp = """\
- a
...
- b
...
"""
docs = list(round_trip_load_all(inp, version=(1, 1)))
assert docs == [['a'], ['b']] # not True, but not reached
|