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

GoogleCloudPlatform · Jan 5, 2023 · a108244 · a108244
1 parent 6f3cbee
commit a108244
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/documentai/snippets/process_document_splitter_sample_test.py b/documentai/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",