@mock.patch('airflow.providers.google.cloud.hooks.bigquery.BigQueryHook.insert_job')
def test_run_query_with_arg(self, mock_insert):
    self.hook.run_query(sql='select 1', destination_dataset_table=f'{DATASET_ID}.{TABLE_ID}', time_partitioning={'type': 'DAY', 'field': 'test_field', 'expirationMs': 1000})
    configuration = {'query': {'query': 'select 1', 'priority': 'INTERACTIVE', 'useLegacySql': True, 'timePartitioning': {'type': 'DAY', 'field': 'test_field', 'expirationMs': 1000}, 'schemaUpdateOptions': [], 'destinationTable': {'projectId': PROJECT_ID, 'datasetId': DATASET_ID, 'tableId': TABLE_ID}, 'allowLargeResults': False, 'flattenResults': None, 'writeDisposition': 'WRITE_EMPTY', 'createDisposition': 'CREATE_IF_NEEDED'}}
    mock_insert.assert_called_once_with(configuration=configuration, project_id=PROJECT_ID)
@mock.patch('airflow.providers.google.cloud.hooks.bigquery.BigQueryHook.insert_job')
def test_run_query_with_arg(self, mock_insert):
    self.hook.run_query(sql='select 1', destination_dataset_table=f'{DATASET_ID}.{TABLE_ID}', time_partitioning={'type': 'DAY', 'field': 'test_field', 'expirationMs': 1000})
    configuration = {'query': {'query': 'select 1', 'priority': 'INTERACTIVE', 'useLegacySql': True, 'timePartitioning': {'type': 'DAY', 'field': 'test_field', 'expirationMs': 1000}, 'schemaUpdateOptions': [], 'destinationTable': {'projectId': PROJECT_ID, 'datasetId': DATASET_ID, 'tableId': TABLE_ID}, 'allowLargeResults': False, 'flattenResults': None, 'writeDisposition': 'WRITE_EMPTY', 'createDisposition': 'CREATE_IF_NEEDED'}}
    mock_insert.assert_called_once_with(configuration=configuration, project_id=PROJECT_ID)