def test_deprecation_warning(self, mock_hook): with self.assertWarns(DeprecationWarning) as warning: DataprocSubmitSparkJobOperator( task_id=TASK_ID, region=GCP_LOCATION, main_class=self.main_class, dataproc_jars=self.jars ) assert_warning("DataprocSubmitJobOperator", warning)