Fix broken test
(Update snapshot to be correct.)
This commit is contained in:
parent
276449287d
commit
8a80bcad64
1 changed files with 29 additions and 14 deletions
|
|
@ -97,9 +97,11 @@ def flatten_document(doc: wadler.Document, src: str) -> list:
|
||||||
case wadler.NewLine():
|
case wadler.NewLine():
|
||||||
return ["<newline>"]
|
return ["<newline>"]
|
||||||
case wadler.Indent():
|
case wadler.Indent():
|
||||||
return [f"<indent {doc.amount}>", flatten_document(doc.doc, src)]
|
return [[f"<indent {doc.amount}>", flatten_document(doc.doc, src)]]
|
||||||
case wadler.Text(start, end):
|
case wadler.Text(start, end):
|
||||||
return [src[start:end]]
|
return [src[start:end]]
|
||||||
|
case wadler.Literal(text):
|
||||||
|
return [text]
|
||||||
case wadler.Group():
|
case wadler.Group():
|
||||||
return [flatten_document(doc.child, src)]
|
return [flatten_document(doc.child, src)]
|
||||||
case wadler.Lazy():
|
case wadler.Lazy():
|
||||||
|
|
@ -125,23 +127,36 @@ def test_convert_tree_to_document():
|
||||||
assert doc == [
|
assert doc == [
|
||||||
[
|
[
|
||||||
"{",
|
"{",
|
||||||
['"a"', ":", "true"],
|
|
||||||
",",
|
|
||||||
"<newline>",
|
|
||||||
[
|
[
|
||||||
'"b"',
|
"<indent 1>",
|
||||||
":",
|
|
||||||
[
|
[
|
||||||
"[",
|
"<newline>",
|
||||||
"1",
|
['"a"', ":", " ", "true"],
|
||||||
",",
|
",",
|
||||||
"<newline>",
|
"<newline>",
|
||||||
"2",
|
[
|
||||||
",",
|
'"b"',
|
||||||
"<newline>",
|
":",
|
||||||
"3",
|
" ",
|
||||||
"<newline>",
|
[
|
||||||
"]",
|
"[",
|
||||||
|
[
|
||||||
|
"<indent 1>",
|
||||||
|
[
|
||||||
|
"<newline>",
|
||||||
|
"1",
|
||||||
|
",",
|
||||||
|
"<newline>",
|
||||||
|
"2",
|
||||||
|
",",
|
||||||
|
"<newline>",
|
||||||
|
"3",
|
||||||
|
],
|
||||||
|
],
|
||||||
|
"<newline>",
|
||||||
|
"]",
|
||||||
|
],
|
||||||
|
],
|
||||||
],
|
],
|
||||||
],
|
],
|
||||||
"<newline>",
|
"<newline>",
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue