@mock.patch('airflow.providers.google.cloud.operators.bigquery.BigQueryHook')
def test_execute(self, mock_hook):
    dataset_resource = {'friendlyName': 'Test DS'}
    operator = BigQueryUpdateDatasetOperator(dataset_resource=dataset_resource, task_id=TASK_ID, dataset_id=TEST_DATASET, project_id=TEST_GCP_PROJECT_ID)
    operator.execute(None)
    mock_hook.return_value.update_dataset.assert_called_once_with(dataset_resource=dataset_resource, dataset_id=TEST_DATASET, project_id=TEST_GCP_PROJECT_ID, fields=list(dataset_resource.keys()))
@mock.patch('airflow.providers.google.cloud.operators.bigquery.BigQueryHook')
def test_execute(self, mock_hook):
    dataset_resource = {'friendlyName': 'Test DS'}
    operator = BigQueryUpdateDatasetOperator(dataset_resource=dataset_resource, task_id=TASK_ID, dataset_id=TEST_DATASET, project_id=TEST_GCP_PROJECT_ID)
    operator.execute(None)
    mock_hook.return_value.update_dataset.assert_called_once_with(dataset_resource=dataset_resource, dataset_id=TEST_DATASET, project_id=TEST_GCP_PROJECT_ID, fields=list(dataset_resource.keys()))