Skip to content

Commit

Permalink
Fix flake8
Browse files Browse the repository at this point in the history
  • Loading branch information
KostyaEsmukov committed May 10, 2020
1 parent 9445304 commit ea3d3b8
Show file tree
Hide file tree
Showing 2 changed files with 21 additions and 8 deletions.
5 changes: 1 addition & 4 deletions geopy/geocoders/algolia.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,4 @@
from geopy.compat import (
Request,
urlencode,
)
from geopy.compat import Request, urlencode
from geopy.geocoders.base import DEFAULT_SENTINEL, Geocoder
from geopy.location import Location
from geopy.util import logger
Expand Down
24 changes: 20 additions & 4 deletions test/geocoders/algolia.py
Original file line number Diff line number Diff line change
@@ -1,15 +1,18 @@
import unittest

from geopy.geocoders import AlgoliaPlaces
from test.geocoders.tomtom import BaseTomTomTestCase
from test.geocoders.util import GeocoderTestBase, env


@unittest.skipUnless(
bool(env.get('ALGOLIA_PLACES_APP_ID')) and bool(env.get('ALGOLIA_PLACES_API_KEY')),
'No ALGOLIA_PLACES_APP_ID and/or no ALGOLIA_PLACES_API_KEY env variables setted'
)
class AlgoliaPlacesTestCase(BaseTomTomTestCase, GeocoderTestBase):
class AlgoliaPlacesTestCase(GeocoderTestBase):

@classmethod
def setUpClass(cls):
cls.geocoder = cls.make_geocoder()

@classmethod
def make_geocoder(cls, **kwargs):
Expand All @@ -19,6 +22,19 @@ def make_geocoder(cls, **kwargs):
timeout=3,
**kwargs)

def test_user_agent_custom(self):
geocoder = self.make_geocoder(
user_agent='my_user_agent/1.0'
)
self.assertEqual(geocoder.headers['User-Agent'], 'my_user_agent/1.0')

def test_geocode(self):
location = self.geocode_run(
{'query': 'москва'},
{'latitude': 55.75587, 'longitude': 37.61768},
)
self.assertIn('Москва', location.address)

def test_reverse(self):
location = self.reverse_run(
{'query': '51, -0.13', 'language': 'en'},
Expand Down Expand Up @@ -53,15 +69,15 @@ def test_limit(self):

def test_countries(self):
countries = ["ES"]
locations = self.geocode_run(
self.geocode_run(
{'query': 'Madrid', 'language': 'en',
'exactly_one': False, 'countries': countries},
{},
)

def test_countries_no_result(self):
countries = ["NO", "IT"]
locations = self.geocode_run(
self.geocode_run(
{'query': 'Madrid', 'language': 'en',
'exactly_one': False, 'countries': countries},
{},
Expand Down

0 comments on commit ea3d3b8

Please sign in to comment.