Updates tests for clarity

Author: chalmerlowe

google/cloud/bigquery/dataset.py

@@ -532,6 +532,7 @@ class Dataset(object):
         "default_rounding_mode": "defaultRoundingMode",
         "resource_tags": "resourceTags",
         "external_catalog_dataset_options": "externalCatalogDatasetOptions",
+        "access_policy_version": "accessPolicyVersion",
     }
 
     def __init__(self, dataset_ref) -> None:
@@ -922,6 +923,16 @@ def external_catalog_dataset_options(self, value):
             self._PROPERTY_TO_API_FIELD["external_catalog_dataset_options"]
         ] = (value.to_api_repr() if value is not None else None)
 
+    @property
+    def access_policy_version(self):
+        return self._properties.get("accessPolicyVersion")
+
+    @access_policy_version.setter
+    def access_policy_version(self, value):
+        if not isinstance(value, int) and value is not None:
+            raise ValueError("Pass an integer, or None")
+        self._properties[self._PROPERTY_TO_API_FIELD["access_policy_version"]] = value
+
     @classmethod
     def from_string(cls, full_dataset_id: str) -> "Dataset":
         """Construct a dataset from fully-qualified dataset ID.

tests/unit/test_dataset.py

@@ -796,6 +796,7 @@ def test_ctor_defaults(self):
         self.assertIsNone(dataset.friendly_name)
         self.assertIsNone(dataset.location)
         self.assertEqual(dataset.is_case_insensitive, False)
+        self.assertIsNone(dataset.access_policy_version)
 
     def test_ctor_string(self):
         dataset = self._make_one("some-project.some_dset")
@@ -1152,6 +1153,33 @@ def test_external_catalog_dataset_options_to_api_repr(self):
         expected = api_repr["externalCatalogDatasetOptions"]
         assert result == expected
 
+    def test_access_policy_version_valid_input(self):
+        dataset = self._make_one(self.DS_REF)
+        # Valid inputs for access_policy_version are currently
+        # ints 1, 2, 3, and None
+        for expected in [1, 2, 3, None]:
+            # set property using setter and integer
+            dataset.access_policy_version = expected
+
+            # check getter and _properties dict
+            assert (
+                dataset.access_policy_version == expected
+            ), f"Expected {expected} but got {dataset.access_policy_version}"
+            assert dataset._properties["accessPolicyVersion"] == expected
+
+    def test_access_policy_version_invalid_input(self):
+        dataset = self._make_one(self.DS_REF)
+        # Valid inputs for access_policy_version are currently
+        # ints 1, 2, 3, and None
+
+        with pytest.raises(ValueError):
+            invalid_value = "a string"
+            dataset.access_policy_version = invalid_value
+
+        with pytest.raises(ValueError):
+            invalid_value = 42.0
+            dataset.access_policy_version = invalid_value
+
 
 class TestDatasetListItem(unittest.TestCase):
     @staticmethod