@mock.patch('airflow.providers.google.cloud.hooks.bigquery.BigQueryHook.get_service')
@mock.patch('airflow.providers.google.cloud.hooks.bigquery.BigQueryHook.insert_job')
def test_location_propagates_properly(self, run_with_config, _):
    assert (self.hook.location is None)
    self.hook.run_query(sql='select 1', location='US')
    assert (run_with_config.call_count == 1)
    assert (self.hook.location == 'US')
@mock.patch('airflow.providers.google.cloud.hooks.bigquery.BigQueryHook.get_service')
@mock.patch('airflow.providers.google.cloud.hooks.bigquery.BigQueryHook.insert_job')
def test_location_propagates_properly(self, run_with_config, _):
    assert (self.hook.location is None)
    self.hook.run_query(sql='select 1', location='US')
    assert (run_with_config.call_count == 1)
    assert (self.hook.location == 'US')