Fix broken test

(Update snapshot to be correct.)
This commit is contained in:
John Doty 2024-09-11 11:28:53 -07:00
parent 276449287d
commit 8a80bcad64

View file

@ -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>",