Remove non async methods (#69)

* Remove non async methods

* More cleanup / updated tests

* Use annotation for tests

* Add pytest-asyncio dependency

* await

* Marked def with `async`

* Bump version

Author: Donatas-L

setup.cfg

@@ -26,6 +26,7 @@ install_requires =
 	protobuf == 4.21.12
 test_require =
 	pytest
+	pytest-asyncio
 
 [flake8]
 per-file-ignores = __init__.py:F401

tests/geocoding_test.py

@@ -1,11 +1,16 @@
-def test_geocoding_search(sdk):
-    response = sdk.geocoding(
+import pytest
+
+
+@pytest.mark.asyncio
+async def test_geocoding_search(sdk):
+    response = await sdk.geocoding_async(
         query="Parliament square", limit=30, within_countries=["gb", "de"]
     )
     assert len(response.features) > 0
     assert len(response.features) < 31
 
 
-def test_geocoding_reverse(sdk):
-    response = sdk.geocoding_reverse(lat=51.507281, lng=-0.132120)
+@pytest.mark.asyncio
+async def test_geocoding_reverse(sdk):
+    response = await sdk.geocoding_reverse_async(lat=51.507281, lng=-0.132120)
     assert len(response.features) > 0

tests/map_info_test.py

@@ -1,3 +1,7 @@
-def test_map_info(sdk):
-    maps = sdk.map_info()
+import pytest
+
+
+@pytest.mark.asyncio
+async def test_map_info(sdk):
+    maps = await sdk.map_info_async()
     assert len(maps) > 0

tests/postcodes_test.py

@@ -1,55 +1,62 @@
+import pytest
 from datetime import datetime
 
 from traveltimepy import Coordinates, PublicTransport
 
 
-def test_departures(sdk):
-    results = sdk.postcodes(
+@pytest.mark.asyncio
+async def test_departures(sdk):
+    results = await sdk.postcodes_async(
         coordinates=[Coordinates(lat=51.507609, lng=-0.128315)],
         departure_time=datetime.now(),
         transportation=PublicTransport(),
     )
     assert len(results) > 0
 
 
-def test_arrivals(sdk):
-    results = sdk.postcodes(
+@pytest.mark.asyncio
+async def test_arrivals(sdk):
+    results = await sdk.postcodes_async(
         coordinates=[Coordinates(lat=51.507609, lng=-0.128315)],
         arrival_time=datetime.now(),
         transportation=PublicTransport(),
     )
     assert len(results) > 0
 
 
-def test_districts_departure(sdk):
-    results = sdk.postcode_districts(
+@pytest.mark.asyncio
+async def test_districts_departure(sdk):
+    results = await sdk.postcodes_districts_async(
         coordinates=[Coordinates(lat=51.507609, lng=-0.128315)],
         departure_time=datetime.now(),
         transportation=PublicTransport(),
     )
     assert len(results) > 0
 
 
-def test_districts_arrival(sdk):
-    results = sdk.postcode_districts(
+@pytest.mark.asyncio
+async def test_districts_arrival(sdk):
+    results = await sdk.postcodes_districts_async(
         coordinates=[Coordinates(lat=51.507609, lng=-0.128315)],
         arrival_time=datetime.now(),
         transportation=PublicTransport(),
     )
     assert len(results) > 0
 
 
-def test_sectors_departure(sdk):
-    results = sdk.postcode_sectors(
+@pytest.mark.asyncio
+async def test_sectors_departure(sdk):
+    results = await sdk.postcodes_sectors_async(
         coordinates=[Coordinates(lat=51.507609, lng=-0.128315)],
         departure_time=datetime.now(),
         transportation=PublicTransport(),
     )
     assert len(results) > 0
 
 
-def test_sectors_arrival(sdk):
-    results = sdk.postcode_sectors(
+@pytest.mark.asyncio
+async def test_sectors_arrival(sdk):
+    results = await sdk.postcodes_sectors_async(
         coordinates=[Coordinates(lat=51.507609, lng=-0.128315)],
         arrival_time=datetime.now(),
         transportation=PublicTransport(),

tests/routes_test.py

@@ -1,10 +1,12 @@
+import pytest
 from datetime import datetime
 
 from traveltimepy import PublicTransport
 
 
-def test_departures(sdk, locations):
-    results = sdk.routes(
+@pytest.mark.asyncio
+async def test_departures(sdk, locations):
+    results = await sdk.routes_async(
         locations=locations,
         search_ids={
             "London center": ["Hyde Park", "ZSL London Zoo"],
@@ -16,8 +18,9 @@ def test_departures(sdk, locations):
     assert len(results) == 2
 
 
-def test_arrivals(sdk, locations):
-    results = sdk.routes(
+@pytest.mark.asyncio
+async def test_arrivals(sdk, locations):
+    results = await sdk.routes_async(
         locations=locations,
         search_ids={
             "London center": ["Hyde Park", "ZSL London Zoo"],

tests/supported_locations.py

@@ -1,14 +1,17 @@
+import pytest
+
 from traveltimepy import Location, Coordinates
 
 
-def test_supported_locations(sdk):
+@pytest.mark.asyncio
+async def test_supported_locations(sdk):
     locations = [
         Location(id="Kaunas", coords=Coordinates(lat=54.900008, lng=23.957734)),
         Location(id="London", coords=Coordinates(lat=51.506756, lng=-0.12805)),
         Location(id="Bangkok", coords=Coordinates(lat=13.761866, lng=100.544818)),
         Location(id="Lisbon", coords=Coordinates(lat=38.721869, lng=-9.138549)),
         Location(id="Unsupported", coords=Coordinates(lat=68.721869, lng=-9.138549)),
     ]
-    response = sdk.supported_locations(locations)
+    response = await sdk.supported_locations_async(locations)
     assert len(response.locations) == 4
     assert len(response.unsupported_locations) == 1

tests/time_filter_fast_test.py

@@ -1,8 +1,11 @@
+import pytest
+
 from traveltimepy.dto.requests.time_filter_fast import Transportation
 
 
-def test_one_to_many(sdk, locations):
-    results = sdk.time_filter_fast(
+@pytest.mark.asyncio
+async def test_one_to_many(sdk, locations):
+    results = await sdk.time_filter_fast_async(
         locations=locations,
         search_ids={
             "London center": ["Hyde Park", "ZSL London Zoo"],
@@ -14,8 +17,9 @@ def test_one_to_many(sdk, locations):
     assert len(results) > 0
 
 
-def test_many_to_one(sdk, locations):
-    results = sdk.time_filter_fast(
+@pytest.mark.asyncio
+async def test_many_to_one(sdk, locations):
+    results = await sdk.time_filter_fast_async(
         locations=locations,
         search_ids={
             "London center": ["Hyde Park", "ZSL London Zoo"],

tests/time_filter_test.py

@@ -1,10 +1,12 @@
+import pytest
 from datetime import datetime
 
 from traveltimepy import PublicTransport
 
 
-def test_departures(sdk, locations):
-    results = sdk.time_filter(
+@pytest.mark.asyncio
+async def test_departures(sdk, locations):
+    results = await sdk.time_filter_async(
         locations=locations,
         search_ids={
             "London center": ["Hyde Park", "ZSL London Zoo"],
@@ -16,8 +18,9 @@ def test_departures(sdk, locations):
     assert len(results) == 2
 
 
-def test_arrivals(sdk, locations):
-    results = sdk.time_filter(
+@pytest.mark.asyncio
+async def test_arrivals(sdk, locations):
+    results = await sdk.time_filter_async(
         locations=locations,
         search_ids={
             "London center": ["Hyde Park", "ZSL London Zoo"],

tests/time_map_test.py

@@ -1,10 +1,12 @@
+import pytest
 from datetime import datetime
 
 from traveltimepy import Coordinates, Driving
 
 
-def test_departures(sdk):
-    results = sdk.time_map(
+@pytest.mark.asyncio
+async def test_departures(sdk):
+    results = await sdk.time_map_async(
         coordinates=[
             Coordinates(lat=51.507609, lng=-0.128315),
             Coordinates(lat=51.517609, lng=-0.138315),
@@ -16,8 +18,9 @@ def test_departures(sdk):
     assert len(results) == 2
 
 
-def test_arrivals(sdk):
-    results = sdk.time_map(
+@pytest.mark.asyncio
+async def test_arrivals(sdk):
+    results = await sdk.time_map_async(
         coordinates=[
             Coordinates(lat=51.507609, lng=-0.128315),
             Coordinates(lat=51.517609, lng=-0.138315),
@@ -29,8 +32,9 @@ def test_arrivals(sdk):
     assert len(results) == 2
 
 
-def test_union_departures(sdk):
-    result = sdk.union(
+@pytest.mark.asyncio
+async def test_union_departures(sdk):
+    result = await sdk.union_async(
         coordinates=[
             Coordinates(lat=51.507609, lng=-0.128315),
             Coordinates(lat=51.517609, lng=-0.138315),
@@ -42,8 +46,9 @@ def test_union_departures(sdk):
     assert len(result.shapes) > 0
 
 
-def test_intersection_arrivals(sdk):
-    result = sdk.intersection(
+@pytest.mark.asyncio
+async def test_intersection_arrivals(sdk):
+    result = await sdk.intersection_async(
         coordinates=[
             Coordinates(lat=51.507609, lng=-0.128315),
             Coordinates(lat=51.517609, lng=-0.138315),

traveltimepy/http.py

@@ -77,18 +77,6 @@ def _window_size(rate_limit: int):
         return rate_limit
 
 
-def send_post(
-    response_class: Type[T],
-    path: str,
-    headers: Dict[str, str],
-    request: TravelTimeRequest,
-    sdk_params: SdkParams,
-) -> T:
-    return asyncio.run(
-        send_post_async(response_class, path, headers, request, sdk_params)
-    )
-
-
 async def send_get_async(
     response_class: Type[T],
     path: str,
@@ -109,18 +97,6 @@ async def send_get_async(
                 return await _process_response(response_class, resp)
 
 
-def send_get(
-    response_class: Type[T],
-    path: str,
-    headers: Dict[str, str],
-    sdk_params: SdkParams,
-    params: Dict[str, str] = None,
-) -> T:
-    return asyncio.run(
-        send_get_async(response_class, path, headers, sdk_params, params)
-    )
-
-
 async def _process_response(response_class: Type[T], response: ClientResponse) -> T:
     text = await response.text()
     if response.status != 200: