diff --git a/documentai/snippets/disable_processor_sample_test.py b/documentai/snippets/disable_processor_sample_test.py index 6cd98479d164..8f74e3976396 100644 --- a/documentai/snippets/disable_processor_sample_test.py +++ b/documentai/snippets/disable_processor_sample_test.py @@ -29,7 +29,7 @@ def test_disable_processor(capsys): ) out, _ = capsys.readouterr() - assert "projects" in out or "DISABLED" in out + assert "projects" in out or "Processor" in out # Re-Enable Processor enable_processor_sample.enable_processor_sample( diff --git a/documentai/snippets/enable_processor_sample_test.py b/documentai/snippets/enable_processor_sample_test.py index 0374a6423561..764550cb6c69 100644 --- a/documentai/snippets/enable_processor_sample_test.py +++ b/documentai/snippets/enable_processor_sample_test.py @@ -29,7 +29,7 @@ def test_enable_processor(capsys): ) out, _ = capsys.readouterr() - assert "projects" in out or "ENABLED" in out + assert "projects" in out or "Processor" in out # Re-Disable Processor disable_processor_sample.disable_processor_sample(