test: strip quotes and newlines from output (#279)

busunkim96 · web-flow · commit c0e07a67bce2 · 2022-02-25T19:23:38.000-05:00
diff --git a/document_ai/snippets/process_document_splitter_sample_test.py b/document_ai/snippets/process_document_splitter_sample_test.py
@@ -33,6 +33,9 @@ def test_process_documents(capsys):
     )
     out, _ = capsys.readouterr()
 
+    # Remove newlines and quotes from output for easier comparison
+    out = out.replace(' "" ', " ").replace("\n", "")
+
     expected_strings = [
         "Found 8 subdocuments",
         "confident that pages 1 to 2 are a subdocument",

Original file line number	Diff line number	Diff line change
`@@ -33,6 +33,9 @@ def test_process_documents(capsys):`
`33`	`33`	`)`
`34`	`34`	`out, _ = capsys.readouterr()`
`35`	`35`
	`36`	`+ # Remove newlines and quotes from output for easier comparison`
	`37`	`+ out = out.replace(' "" ', " ").replace("\n", "")`
	`38`	`+`
`36`	`39`	`expected_strings = [`
`37`	`40`	`"Found 8 subdocuments",`
`38`	`41`	`"confident that pages 1 to 2 are a subdocument",`