test: Added conditional skip

tests/integration_tests/skip_helpers.py

@@ -0,0 +1,9 @@
+import os
+
+
+def should_skip_bedrock_integration_tests() -> bool:
+    return os.getenv("AWS_ACCESS_KEY_ID") is None and os.getenv("AWS_SECRET_ACCESS_KEY") is None
+
+
+def should_skip_studio_integration_tests() -> bool:
+    return os.getenv("AI21_API_KEY") is None

tests/integration_tests/test_bedrock.py

@@ -1,18 +1,20 @@
 """
 Run this script after setting the environment variable called AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY
 """
-
+import subprocess
 from pathlib import Path
 
 import pytest
-import subprocess
+
+from tests.integration_tests.skip_helpers import should_skip_bedrock_integration_tests
 
 BEDROCK_DIR = "bedrock"
 
 
 BEDROCK_PATH = Path(__file__).parent.parent.parent / "examples" / BEDROCK_DIR
 
 
+@pytest.mark.skipif(should_skip_bedrock_integration_tests(), reason="No keys supplied for AWS. Skipping.")
 @pytest.mark.parametrize(
     argnames=["test_file_name"],
     argvalues=[

tests/integration_tests/test_studio.py

@@ -7,9 +7,12 @@
 import pytest
 import subprocess
 
+from tests.integration_tests.skip_helpers import should_skip_studio_integration_tests
+
 STUDIO_PATH = Path(__file__).parent.parent.parent / "examples" / "studio"
 
 
+@pytest.mark.skipif(should_skip_studio_integration_tests(), reason="No key supplied for AI21 Studio. Skipping.")
 @pytest.mark.parametrize(
     argnames=["test_file_name"],
     argvalues=[