add tests

Author: tswast

pandas_gbq/core/resource_references.py

@@ -48,6 +48,9 @@ def parse_table_id(table_id: str) -> BigLakeTableId | BigQueryTableId:
         raise ValueError(f"Invalid table ID: {table_id}")
 
     inner_parts = regex_match.group("inner_parts").split(".")
+    if any(part == "" for part in inner_parts):
+        raise ValueError(f"Invalid table ID: {table_id}")
+
     if len(inner_parts) == 1:
         return BigQueryTableId(
             project_id=regex_match.group("project"),

tests/unit/core/test_core_resource_references.py

@@ -2,3 +2,73 @@
 # Use of this source code is governed by a BSD-style
 # license that can be found in the LICENSE file.
 
+import pytest
+
+from pandas_gbq.core import resource_references
+
+
+@pytest.mark.parametrize(
+    ["table_id", "expected"],
+    [
+        (
+            "my-project.my_dataset.my_table",
+            resource_references.BigQueryTableId(
+                project_id="my-project",
+                dataset_id="my_dataset",
+                table_id="my_table",
+            ),
+        ),
+        (
+            "google.com:my-project.my_dataset.my_table",
+            resource_references.BigQueryTableId(
+                project_id="my-project",
+                dataset_id="my_dataset",
+                table_id="my_table",
+            ),
+        ),
+        (
+            "my-project.my_catalog.my_table",
+            resource_references.BigQueryTableId(
+                project_id="my-project",
+                dataset_id="my_catalog",
+                table_id="my_table",
+            ),
+        ),
+        (
+            "my-project.my_catalog.my_namespace.my_table",
+            resource_references.BigLakeTableId(
+                project="my-project",
+                catalog="my_catalog",
+                namespace=("my_namespace",),
+                table="my_table",
+            ),
+        ),
+        (
+            "my-project.my_catalog.my_namespace1.my_namespace2.my_table",
+            resource_references.BigLakeTableId(
+                project="my-project",
+                catalog="my_catalog",
+                namespace=("my_namespace1", "my_namespace2"),
+                table="my_table",
+            ),
+        ),
+    ],
+)
+def test_parse_table_id_valid(table_id, expected):
+    result = resource_references.parse_table_id(table_id)
+    assert result == expected
+
+
+@pytest.mark.parametrize(
+    "table_id",
+    [
+        "my-project",
+        "my-project.my_dataset",
+        ".my_dataset.my_table",
+        "my-project.my_dataset.",
+        "my-project..my_table",
+    ],
+)
+def test_parse_table_id_invalid(table_id):
+    with pytest.raises(ValueError):
+        resource_references.parse_table_id(table_id)