From f679e1c76b5c0dd6fd46d8743e4f65795400e02d Mon Sep 17 00:00:00 2001 From: Logan Date: Sun, 14 Apr 2024 15:12:22 -0600 Subject: [PATCH] Skip tests if CICD doesn't populate the secrets (#142) --- tests/test_reader.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/tests/test_reader.py b/tests/test_reader.py index e1dfab4..2736e99 100644 --- a/tests/test_reader.py +++ b/tests/test_reader.py @@ -1,7 +1,12 @@ import os +import pytest from llama_parse import LlamaParse +@pytest.mark.skipif( + os.environ.get("LLAMA_CLOUD_API_KEY", "") == "", + reason="LLAMA_CLOUD_API_KEY not set", +) def test_simple_page_text() -> None: parser = LlamaParse(result_type="text") @@ -13,6 +18,10 @@ def test_simple_page_text() -> None: assert len(result[0].text) > 0 +@pytest.mark.skipif( + os.environ.get("LLAMA_CLOUD_API_KEY", "") == "", + reason="LLAMA_CLOUD_API_KEY not set", +) def test_simple_page_markdown() -> None: parser = LlamaParse(result_type="markdown") @@ -24,6 +33,10 @@ def test_simple_page_markdown() -> None: assert len(result[0].text) > 0 +@pytest.mark.skipif( + os.environ.get("LLAMA_CLOUD_API_KEY", "") == "", + reason="LLAMA_CLOUD_API_KEY not set", +) def test_simple_page_progress_workers() -> None: parser = LlamaParse(result_type="markdown", show_progress=True, verbose=True)