@mock.patch('airflow.providers.google.cloud.hooks.bigquery.BigQueryHook.insert_job')
def test_run_load_with_arg(self, mock_insert):
    self.hook.run_load(destination_project_dataset_table='my_dataset.my_table', schema_fields=[], source_uris=[], cluster_fields=['field1', 'field2'], time_partitioning={'type': 'DAY'})
    (_, kwargs) = mock_insert.call_args
    assert (kwargs['configuration']['load']['clustering'] == {'fields': ['field1', 'field2']})
@mock.patch('airflow.providers.google.cloud.hooks.bigquery.BigQueryHook.insert_job')
def test_run_load_with_arg(self, mock_insert):
    self.hook.run_load(destination_project_dataset_table='my_dataset.my_table', schema_fields=[], source_uris=[], cluster_fields=['field1', 'field2'], time_partitioning={'type': 'DAY'})
    (_, kwargs) = mock_insert.call_args
    assert (kwargs['configuration']['load']['clustering'] == {'fields': ['field1', 'field2']})