chore: add BigQuery locations to bigframes/constants.py (#578)

* chore: add BigQuery locations to bigframes/constants.py

* Apply suggestions from code review

---------

Co-authored-by: Tim Sweña (Swast) <swast@google.com>

Author: DevStephanie

bigframes/constants.py

@@ -27,3 +27,68 @@
 ABSTRACT_METHOD_ERROR_MESSAGE = f"Abstract method. You have likely encountered a bug. Please share this stacktrace and how you reached it with the BigQuery DataFrames team. {FEEDBACK_LINK}"
 
 DEFAULT_EXPIRATION = datetime.timedelta(days=7)
+
+# https://cloud.google.com/bigquery/docs/locations
+ALL_BIGQUERY_LOCATIONS = frozenset(
+    {
+        "us-east5",
+        "us-south1",
+        "us-central1",
+        "us-west4",
+        "us-west2",
+        "northamerica-northeast1",
+        "us-east4",
+        "us-west1",
+        "us-west3",
+        "southamerica-east1",
+        "southamerica-west1",
+        "us-east1",
+        "northamerica-northeast2",
+        "asia-south2",
+        "asia-east2",
+        "asia-southeast2",
+        "australia-southeast2",
+        "asia-south1",
+        "asia-northeast2",
+        "asia-northeast3",
+        "asia-southeast1",
+        "australia-southeast1",
+        "asia-east1",
+        "asia-northeast1",
+        "europe-west1",
+        "europe-west10",
+        "europe-north1",
+        "europe-west3",
+        "europe-west2",
+        "europe-southwest1",
+        "europe-west8",
+        "europe-west4",
+        "europe-west9",
+        "europe-west12",
+        "europe-central2",
+        "europe-west6",
+        "me-central2",
+        "me-central1",
+        "me-west1",
+        "me-central2",
+        "me-central1",
+        "me-west1",
+        "africa-south1",
+    }
+)
+
+# https://cloud.google.com/storage/docs/regional-endpoints
+REP_ENABLED_BIGQUERY_LOCATIONS = frozenset(
+    {
+        "me-central2",
+        "europe-west9",
+        "europe-west3",
+        "us-east4",
+        "us-west1",
+    }
+)
+
+# https://cloud.google.com/storage/docs/locational-endpoints
+LEP_ENABLED_BIGQUERY_LOCATIONS = frozenset(
+    ALL_BIGQUERY_LOCATIONS - REP_ENABLED_BIGQUERY_LOCATIONS
+)

tests/config.py

@@ -18,8 +18,8 @@
 import pytest
 
 import bigframes
+import bigframes.constants
 import bigframes.session.clients
-from tests import config
 
 
 def _assert_bq_execution_location(session: bigframes.Session):
@@ -66,7 +66,11 @@ def test_bq_location_default():
     _assert_bq_execution_location(session)
 
 
-@pytest.mark.parametrize("bigquery_location", config.ALL_BIGQUERY_LOCATIONS)
+@pytest.mark.parametrize(
+    "bigquery_location",
+    # Sort the set to avoid nondeterminism.
+    sorted(bigframes.constants.ALL_BIGQUERY_LOCATIONS),
+)
 def test_bq_location(bigquery_location):
     session = bigframes.Session(
         context=bigframes.BigQueryOptions(location=bigquery_location)
@@ -85,7 +89,8 @@ def test_bq_location(bigquery_location):
 
 @pytest.mark.parametrize(
     "bigquery_location",
-    config.REP_ENABLED_BIGQUERY_LOCATIONS,
+    # Sort the set to avoid nondeterminism.
+    sorted(bigframes.constants.REP_ENABLED_BIGQUERY_LOCATIONS),
 )
 def test_bq_rep_endpoints(bigquery_location):
     session = bigframes.Session(
@@ -108,7 +113,8 @@ def test_bq_rep_endpoints(bigquery_location):
 
 @pytest.mark.parametrize(
     "bigquery_location",
-    config.LEP_ENABLED_BIGQUERY_LOCATIONS,
+    # Sort the set to avoid nondeterminism.
+    sorted(bigframes.constants.LEP_ENABLED_BIGQUERY_LOCATIONS),
 )
 def test_bq_lep_endpoints(bigquery_location):
     # We are not testing BigFrames Session for LEP endpoints because it involves