xinbinhuang commented on a change in pull request #15250:
URL: https://github.com/apache/airflow/pull/15250#discussion_r615353197
##########
File path: tests/providers/google/cloud/operators/test_dataproc.py
##########
@@ -144,6 +144,26 @@ def test_image_version(self):
)
assert "custom_image and image_version" in str(ctx.value)
+ def test_custom_image_family_1(self):
+ with pytest.raises(ValueError) as ctx:
+ ClusterGenerator(
+ image_version="image_version",
+ custom_image_family="custom_image_family",
+ project_id=GCP_PROJECT,
+ cluster_name=CLUSTER_NAME,
+ )
+ assert "image_version and custom_image_family" in str(ctx.value)
+
+ def test_custom_image_family_2(self):
Review comment:
```suggestion
def test_custom_image_family_error_with_custom_image(self):
```
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
