remove markers for test and only run one test per cloudbuild

GoogleCloudPlatform · Dec 20, 2023 · d54c1a4 · d54c1a4
1 parent be31df2
commit d54c1a4
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 7 deletions.
diff --git a/retrieval_service/datastore/providers/postgres_test.py b/retrieval_service/datastore/providers/postgres_test.py
@@ -26,7 +26,7 @@
 from .test_data import query_embedding1, query_embedding2, query_embedding3
 from .utils import get_env_var
 
-pytestmark = [pytest.mark.asyncio(scope="module"), pytest.mark.postgres]
+pytestmark = pytest.mark.asyncio(scope="module")
 
 
 @pytest.fixture(scope="module")

diff --git a/retrieval_service/pyproject.toml b/retrieval_service/pyproject.toml
@@ -8,8 +8,3 @@ warn_unused_configs = true
 [[tool.mypy.overrides]]
 module = ["pgvector.asyncpg"]
 ignore_missing_imports = true
-
-[tool.pytest.ini_options]
-markers = [
-    "postgres",
-]
diff --git a/retrieval_service/sql-proxy/cloudsql.tests.cloudbuild.yaml b/retrieval_service/sql-proxy/cloudsql.tests.cloudbuild.yaml
@@ -102,7 +102,7 @@ steps:
         # Set env var expected by tests
         export DB_USER=$$CLOUDSQLUSER
         export DB_PASS=$$CLOUDSQLPASSWORD
-        python -m pytest -m "not postgres"
+        python -m pytest -m ../datastore/providers/cloudsql_postgres.py
 
   - id: Run database export
     name: python:3.11