@mock.patch('airflow.providers.google.cloud.hooks.bigquery.Client')
def test_get_dataset(self, mock_client):
    _expected_result = {'kind': 'bigquery#dataset', 'location': 'US', 'id': 'your-project:dataset_2_test', 'datasetReference': {'projectId': 'your-project', 'datasetId': 'dataset_2_test'}}
    expected_result = Dataset.from_api_repr(_expected_result)
    mock_client.return_value.get_dataset.return_value = expected_result
    result = self.hook.get_dataset(dataset_id=DATASET_ID, project_id=PROJECT_ID)
    mock_client.return_value.get_dataset.assert_called_once_with(dataset_ref=DatasetReference(PROJECT_ID, DATASET_ID))
    assert (result == expected_result)
@mock.patch('airflow.providers.google.cloud.hooks.bigquery.Client')
def test_get_dataset(self, mock_client):
    _expected_result = {'kind': 'bigquery#dataset', 'location': 'US', 'id': 'your-project:dataset_2_test', 'datasetReference': {'projectId': 'your-project', 'datasetId': 'dataset_2_test'}}
    expected_result = Dataset.from_api_repr(_expected_result)
    mock_client.return_value.get_dataset.return_value = expected_result
    result = self.hook.get_dataset(dataset_id=DATASET_ID, project_id=PROJECT_ID)
    mock_client.return_value.get_dataset.assert_called_once_with(dataset_ref=DatasetReference(PROJECT_ID, DATASET_ID))
    assert (result == expected_result)