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

GoogleCloudPlatform · Feb 26, 2022 · c0e07a6 · c0e07a6
1 parent cdf1d20
commit c0e07a6
Showing 1 changed file with 3 additions and 0 deletions.
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",