moves test out of class to accommodate pytest params

Author: chalmerlowe

tests/unit/test_table.py

@@ -6322,72 +6322,6 @@ def test_constructor_explicit_with_none(self):
         self.assertIsNone(table_constraint.primary_key)
         self.assertIsNone(table_constraint.foreign_keys)
 
-    @pytest.mark.parametrize(
-        "self_pk_name,self_fk_name,other_pk_name,other_fk_name,expected_equal",
-        [
-            (None, None, None, None, True),
-            ('pkey', None, 'pkey', None, True),
-            ('pkey', 'fkey', 'pkey', 'fkey', True),
-            (None, 'fkey', None, 'fkey', True),
-            ('pkey', None, 'pkey_no_match', None, False),
-            ('pkey', 'fkey', 'pkey_no_match', 'fkey_no_match', False),
-            (None, 'fkey', None, 'fkey_no_match', False),
-            ('pkey', 'fkey', 'pkey_no_match', 'fkey', False),
-            ('pkey', 'fkey', 'pkey', 'fkey_no_match', False),
-        ]
-    )        
-    def test_eq_parametrized(self, self_pk_name, self_fk_name, other_pk_name, other_fk_name, expected_equal):
-        # Imports are placed here to ensure they are self-contained for this example.
-        # In a real test file, they would likely be at the top of the file.
-        from google.cloud.bigquery.table import (
-            ColumnReference,
-            ForeignKey,
-            PrimaryKey,
-            TableReference,
-            TableConstraints,
-        )
-
-        # Helper function to create a PrimaryKey object or None
-        def _create_primary_key(name):
-            if name is None:
-                return None
-            return PrimaryKey(columns=[name])
-
-        # Helper function to create a list of ForeignKey objects or None
-        def _create_foreign_keys(name):
-            if name is None:
-                return None
-            # Using a generic referenced_table and column_references for simplicity
-            # The 'name' parameter ensures different ForeignKey objects for different names
-            return [
-                ForeignKey(
-                    name=name,
-                    referenced_table=TableReference.from_string(
-                        f"my-project.my-dataset.{name}_referenced_table"
-                    ),
-                    column_references=[
-                        ColumnReference(
-                            referencing_column=f"{name}_ref_col",
-                            referenced_column=f"{name}_pk_col",
-                        )
-                    ],
-                )
-            ]
-
-        # Create the two TableConstraints instances for comparison
-        tc1 = TableConstraints(
-            primary_key=_create_primary_key(self_pk_name),
-            foreign_keys=_create_foreign_keys(self_fk_name),
-        )
-        tc2 = TableConstraints(
-            primary_key=_create_primary_key(other_pk_name),
-            foreign_keys=_create_foreign_keys(other_fk_name),
-        )
-
-        # Assert the equality based on the expected outcome
-        assert (tc1 == tc2) == expected_equal
-
-
     def test__eq__other_type(self):
         from google.cloud.bigquery.table import (
             PrimaryKey,
@@ -6605,6 +6539,74 @@ def test_to_api_repr_empty_constraints(self):
         self.assertEqual(instance.to_api_repr(), expected)
 
 
+@pytest.mark.parametrize(
+    "self_pk_name,self_fk_name,other_pk_name,other_fk_name,expected_equal",
+    [
+        (None, None, None, None, True),
+        ("pkey", None, "pkey", None, True),
+        ("pkey", "fkey", "pkey", "fkey", True),
+        (None, "fkey", None, "fkey", True),
+        ("pkey", None, "pkey_no_match", None, False),
+        ("pkey", "fkey", "pkey_no_match", "fkey_no_match", False),
+        (None, "fkey", None, "fkey_no_match", False),
+        ("pkey", "fkey", "pkey_no_match", "fkey", False),
+        ("pkey", "fkey", "pkey", "fkey_no_match", False),
+    ],
+)
+def test_table_constraint_eq_parametrized(
+    self_pk_name, self_fk_name, other_pk_name, other_fk_name, expected_equal
+):
+    # Imports are placed here to ensure they are self-contained for this example.
+    # In a real test file, they would likely be at the top of the file.
+    from google.cloud.bigquery.table import (
+        ColumnReference,
+        ForeignKey,
+        PrimaryKey,
+        TableReference,
+        TableConstraints,
+    )
+
+    # Helper function to create a PrimaryKey object or None
+    def _create_primary_key(name):
+        if name is None:
+            return None
+        return PrimaryKey(columns=[name])
+
+    # Helper function to create a list of ForeignKey objects or None
+    def _create_foreign_keys(name):
+        if name is None:
+            return None
+        # Using a generic referenced_table and column_references for simplicity
+        # The 'name' parameter ensures different ForeignKey objects for different names
+        return [
+            ForeignKey(
+                name=name,
+                referenced_table=TableReference.from_string(
+                    f"my-project.my-dataset.{name}_referenced_table"
+                ),
+                column_references=[
+                    ColumnReference(
+                        referencing_column=f"{name}_ref_col",
+                        referenced_column=f"{name}_pk_col",
+                    )
+                ],
+            )
+        ]
+
+    # Create the two TableConstraints instances for comparison
+    tc1 = TableConstraints(
+        primary_key=_create_primary_key(self_pk_name),
+        foreign_keys=_create_foreign_keys(self_fk_name),
+    )
+    tc2 = TableConstraints(
+        primary_key=_create_primary_key(other_pk_name),
+        foreign_keys=_create_foreign_keys(other_fk_name),
+    )
+
+    # Assert the equality based on the expected outcome
+    assert (tc1 == tc2) == expected_equal
+
+
 class TestExternalCatalogTableOptions:
     PROJECT = "test-project"
     DATASET_ID = "test_dataset"