@mock.patch(_GEOCODE)
def test_underlying_geocoding_function_called_with_city_country(self, mocked_geocode, test_dataframe):
    mocked_geocode.side_effect = ['cat', 'dog', 'squirrel']
    geocoded_dataframe = geocode_dataframe(test_dataframe)
    expected_dataframe = pd.DataFrame({'index': [0, 1, 2], 'city': ['London', 'Sheffield', 'Brussels'], 'country': ['UK', 'United Kingdom', 'Belgium'], 'coordinates': ['cat', 'dog', 'squirrel']})
    expected_calls = [mock.call(city='London', country='UK'), mock.call(city='Sheffield', country='United Kingdom'), mock.call(city='Brussels', country='Belgium')]
    assert (geocoded_dataframe.to_dict(orient='records') == expected_dataframe.to_dict(orient='records'))
    assert (mocked_geocode.mock_calls == expected_calls)
@mock.patch(_GEOCODE)
def test_underlying_geocoding_function_called_with_city_country(self, mocked_geocode, test_dataframe):
    mocked_geocode.side_effect = ['cat', 'dog', 'squirrel']
    geocoded_dataframe = geocode_dataframe(test_dataframe)
    expected_dataframe = pd.DataFrame({'index': [0, 1, 2], 'city': ['London', 'Sheffield', 'Brussels'], 'country': ['UK', 'United Kingdom', 'Belgium'], 'coordinates': ['cat', 'dog', 'squirrel']})
    expected_calls = [mock.call(city='London', country='UK'), mock.call(city='Sheffield', country='United Kingdom'), mock.call(city='Brussels', country='Belgium')]
    assert (geocoded_dataframe.to_dict(orient='records') == expected_dataframe.to_dict(orient='records'))
    assert (mocked_geocode.mock_calls == expected_calls)