Isort fixes

benchmarks/common.py

@@ -1,7 +1,7 @@
 import random
 from typing import List
 
-from traveltimepy import Location, Coordinates
+from traveltimepy import Coordinates, Location
 
 
 def generate_float(value: float, radius: float) -> float:

benchmarks/time_filter.py

@@ -3,7 +3,7 @@
 from datetime import datetime
 
 from benchmarks.common import generate_locations
-from traveltimepy import TravelTimeSdk, Driving
+from traveltimepy import Driving, TravelTimeSdk
 
 
 async def generate_matrix(size: int):

benchmarks/time_filter_fast.py

@@ -2,7 +2,7 @@
 import time
 
 from benchmarks.common import generate_locations
-from traveltimepy import TravelTimeSdk, Transportation
+from traveltimepy import Transportation, TravelTimeSdk
 
 
 async def generate_matrix(size: int):

benchmarks/time_map.py

@@ -3,7 +3,7 @@
 from datetime import datetime
 
 from benchmarks.common import generate_coordinates
-from traveltimepy import TravelTimeSdk, Driving
+from traveltimepy import Driving, TravelTimeSdk
 
 
 async def generate_isochrones(size: int):

tests/distance_map_test.py

@@ -1,6 +1,7 @@
-import pytest
 from datetime import datetime
 
+import pytest
+
 from traveltimepy import Coordinates, Driving, LevelOfDetail, TravelTimeSdk
 
 

tests/geocoding_test.py

@@ -1,4 +1,5 @@
 import pytest
+
 from traveltimepy import TravelTimeSdk
 
 

tests/postcodes_test.py

@@ -1,6 +1,7 @@
-import pytest
 from datetime import datetime
 
+import pytest
+
 from traveltimepy import Coordinates, PublicTransport, TravelTimeSdk
 
 

tests/routes_test.py

@@ -1,9 +1,9 @@
+from datetime import datetime
 from typing import List
 
 import pytest
-from datetime import datetime
 
-from traveltimepy import PublicTransport, Driving, Location, Coordinates
+from traveltimepy import Coordinates, Driving, Location, PublicTransport
 from traveltimepy.dto.common import SnapPenalty
 from traveltimepy.sdk import TravelTimeSdk
 

tests/supported_locations.py

@@ -1,6 +1,6 @@
 import pytest
 
-from traveltimepy import Location, Coordinates, TravelTimeSdk
+from traveltimepy import Coordinates, Location, TravelTimeSdk
 
 
 @pytest.mark.asyncio

tests/time_filter_proto_test.py

@@ -2,10 +2,8 @@
 
 from traveltimepy import Coordinates
 from traveltimepy.dto.common import PropertyProto
-from traveltimepy.dto.requests.time_filter_proto import (
-    ProtoTransportation,
-    ProtoCountry,
-)
+from traveltimepy.dto.requests.time_filter_proto import (ProtoCountry,
+                                                         ProtoTransportation)
 from traveltimepy.sdk import TravelTimeSdk
 
 

tests/time_filter_test.py

@@ -1,6 +1,7 @@
-import pytest
 from datetime import datetime
 
+import pytest
+
 from traveltimepy import PublicTransport, TravelTimeSdk
 
 

tests/time_map_test.py

@@ -1,7 +1,9 @@
-import pytest
 from datetime import datetime
 
-from traveltimepy import Coordinates, Driving, LevelOfDetail, Range, TravelTimeSdk
+import pytest
+
+from traveltimepy import (Coordinates, Driving, LevelOfDetail, Range,
+                          TravelTimeSdk)
 
 
 @pytest.mark.asyncio

tests/wkt_parsing_test.py

@@ -1,20 +1,11 @@
 import pytest  # noqa
 
 from traveltimepy import Coordinates
-from traveltimepy.wkt import (
-    parse_wkt,
-    PointModel,
-    LineStringModel,
-    PolygonModel,
-    MultiPointModel,
-    MultiLineStringModel,
-    MultiPolygonModel,
-)
-from traveltimepy.wkt.error import (
-    InvalidWKTStringError,
-    NullGeometryError,
-    InvalidGeometryTypeError,
-)
+from traveltimepy.wkt import (LineStringModel, MultiLineStringModel,
+                              MultiPointModel, MultiPolygonModel, PointModel,
+                              PolygonModel, parse_wkt)
+from traveltimepy.wkt.error import (InvalidGeometryTypeError,
+                                    InvalidWKTStringError, NullGeometryError)
 
 point_wkt = "POINT (0 0)"
 line_wkt = "LINESTRING(0 0, 1 1, 2 2)"

tests/wkt_pretty_print_test.py

@@ -1,14 +1,9 @@
 import pytest  # noqa
 
 from traveltimepy import Coordinates
-from traveltimepy.wkt import (
-    PointModel,
-    LineStringModel,
-    PolygonModel,
-    MultiPointModel,
-    MultiLineStringModel,
-    MultiPolygonModel,
-)
+from traveltimepy.wkt import (LineStringModel, MultiLineStringModel,
+                              MultiPointModel, MultiPolygonModel, PointModel,
+                              PolygonModel)
 
 # Updated mock data for LineStringModel with two distinct points
 mock_point1 = PointModel(coordinates=Coordinates(lat=10, lng=20))

traveltimepy/__init__.py

@@ -1,32 +1,16 @@
 """Python sdk for working with traveltime api"""
 
-from traveltimepy.dto.transportation import (
-    PublicTransport,
-    Driving,
-    Ferry,
-    Walking,
-    Cycling,
-    DrivingTrain,
-    CyclingPublicTransport,
-)
-from traveltimepy.dto.requests.time_filter_proto import (
-    ProtoTransportation,
-    ProtoCountry,
-)
-from traveltimepy.dto.common import (
-    Coordinates,
-    Location,
-    Property,
-    PropertyProto,
-    FullRange,
-    Range,
-    Rectangle,
-    LevelOfDetail,
-)
-
-from traveltimepy.sdk import TravelTimeSdk
-from traveltimepy.dto.requests.time_filter_fast import Transportation
+from traveltimepy.dto.common import (Coordinates, FullRange, LevelOfDetail,
+                                     Location, Property, PropertyProto, Range,
+                                     Rectangle)
 from traveltimepy.dto.requests.postcodes_zones import ZonesProperty
+from traveltimepy.dto.requests.time_filter_fast import Transportation
+from traveltimepy.dto.requests.time_filter_proto import (ProtoCountry,
+                                                         ProtoTransportation)
+from traveltimepy.dto.transportation import (Cycling, CyclingPublicTransport,
+                                             Driving, DrivingTrain, Ferry,
+                                             PublicTransport, Walking)
+from traveltimepy.sdk import TravelTimeSdk
 from traveltimepy.version import __version__
 
 __all__ = [

traveltimepy/dto/common.py

@@ -1,10 +1,10 @@
 from datetime import datetime, time
 from enum import Enum
-from typing import List, Union, Optional
+from typing import List, Optional, Union
 
 from pydantic import field_validator
-from typing_extensions import Literal
 from pydantic.main import BaseModel
+from typing_extensions import Literal
 
 
 class Coordinates(BaseModel):

traveltimepy/dto/requests/distance_map.py

@@ -1,25 +1,16 @@
 import typing
 from datetime import datetime
-
 from typing import List, Optional
 
 from pydantic.main import BaseModel
 
-from traveltimepy import (
-    Coordinates,
-    PublicTransport,
-    Driving,
-    Ferry,
-    Walking,
-    Cycling,
-    DrivingTrain,
-    CyclingPublicTransport,
-    LevelOfDetail,
-)
+from traveltimepy import (Coordinates, Cycling, CyclingPublicTransport,
+                          Driving, DrivingTrain, Ferry, LevelOfDetail,
+                          PublicTransport, Walking)
 from traveltimepy.dto.common import SnapPenalty
 from traveltimepy.dto.requests.request import TravelTimeRequest
 from traveltimepy.dto.responses.time_map import TimeMapResponse
-from traveltimepy.itertools import split, flatten
+from traveltimepy.itertools import flatten, split
 
 
 class DepartureSearch(BaseModel):

traveltimepy/dto/requests/postcodes.py

@@ -1,21 +1,15 @@
 from datetime import datetime
-from typing import List, Union, Optional
+from typing import List, Optional, Union
 
 from pydantic import BaseModel
 
-from traveltimepy.dto.common import Coordinates, Property, FullRange
+from traveltimepy.dto.common import Coordinates, FullRange, Property
 from traveltimepy.dto.requests.request import TravelTimeRequest
 from traveltimepy.dto.responses.postcodes import PostcodesResponse
-from traveltimepy.itertools import split, flatten
-from traveltimepy.dto.transportation import (
-    PublicTransport,
-    Driving,
-    Ferry,
-    Walking,
-    Cycling,
-    DrivingTrain,
-    CyclingPublicTransport,
-)
+from traveltimepy.dto.transportation import (Cycling, CyclingPublicTransport,
+                                             Driving, DrivingTrain, Ferry,
+                                             PublicTransport, Walking)
+from traveltimepy.itertools import flatten, split
 
 
 class ArrivalSearch(BaseModel):

traveltimepy/dto/requests/postcodes_zones.py

@@ -1,25 +1,17 @@
 from datetime import datetime
 from enum import Enum
-from typing import List, Union, Optional
+from typing import List, Optional, Union
 
 from pydantic import BaseModel
 
 from traveltimepy.dto.common import Coordinates, FullRange
 from traveltimepy.dto.requests.request import TravelTimeRequest
-from traveltimepy.dto.responses.zones import (
-    PostcodesDistrictsResponse,
-    PostcodesSectorsResponse,
-)
-from traveltimepy.itertools import split, flatten
-from traveltimepy.dto.transportation import (
-    PublicTransport,
-    Driving,
-    Ferry,
-    Walking,
-    Cycling,
-    DrivingTrain,
-    CyclingPublicTransport,
-)
+from traveltimepy.dto.responses.zones import (PostcodesDistrictsResponse,
+                                              PostcodesSectorsResponse)
+from traveltimepy.dto.transportation import (Cycling, CyclingPublicTransport,
+                                             Driving, DrivingTrain, Ferry,
+                                             PublicTransport, Walking)
+from traveltimepy.itertools import flatten, split
 
 
 class ZonesProperty(str, Enum):

traveltimepy/dto/requests/request.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 from abc import ABC, abstractmethod
-from typing import List, TypeVar, Generic
+from typing import Generic, List, TypeVar
 
 from pydantic import BaseModel
 

traveltimepy/dto/requests/routes.py

@@ -3,19 +3,13 @@
 
 from pydantic.main import BaseModel
 
-from traveltimepy.dto.common import Location, Property, FullRange, SnapPenalty
-from traveltimepy.dto.transportation import (
-    PublicTransport,
-    Driving,
-    Ferry,
-    Walking,
-    Cycling,
-    DrivingTrain,
-    CyclingPublicTransport,
-)
+from traveltimepy.dto.common import FullRange, Location, Property, SnapPenalty
 from traveltimepy.dto.requests.request import TravelTimeRequest
 from traveltimepy.dto.responses.routes import RoutesResponse
-from traveltimepy.itertools import split, flatten
+from traveltimepy.dto.transportation import (Cycling, CyclingPublicTransport,
+                                             Driving, DrivingTrain, Ferry,
+                                             PublicTransport, Walking)
+from traveltimepy.itertools import flatten, split
 
 
 class ArrivalSearch(BaseModel):

traveltimepy/dto/requests/supported_locations.py

@@ -2,7 +2,8 @@
 
 from traveltimepy.dto.common import Location
 from traveltimepy.dto.requests.request import TravelTimeRequest
-from traveltimepy.dto.responses.supported_locations import SupportedLocationsResponse
+from traveltimepy.dto.responses.supported_locations import \
+    SupportedLocationsResponse
 from traveltimepy.itertools import flatten
 
 

traveltimepy/dto/requests/time_filter.py

@@ -3,19 +3,13 @@
 
 from pydantic.main import BaseModel
 
-from traveltimepy.dto.common import Location, FullRange, Property, SnapPenalty
+from traveltimepy.dto.common import FullRange, Location, Property, SnapPenalty
 from traveltimepy.dto.requests.request import TravelTimeRequest
 from traveltimepy.dto.responses.time_filter import TimeFilterResponse
-from traveltimepy.itertools import split, flatten
-from traveltimepy.dto.transportation import (
-    PublicTransport,
-    Driving,
-    Ferry,
-    Walking,
-    Cycling,
-    DrivingTrain,
-    CyclingPublicTransport,
-)
+from traveltimepy.dto.transportation import (Cycling, CyclingPublicTransport,
+                                             Driving, DrivingTrain, Ferry,
+                                             PublicTransport, Walking)
+from traveltimepy.itertools import flatten, split
 
 
 class ArrivalSearch(BaseModel):

traveltimepy/dto/requests/time_filter_fast.py

@@ -1,12 +1,12 @@
 from typing import List, Optional
-from typing_extensions import Literal
 
 from pydantic import BaseModel
+from typing_extensions import Literal
 
 from traveltimepy.dto.common import Location, Property, SnapPenalty
 from traveltimepy.dto.requests.request import TravelTimeRequest
 from traveltimepy.dto.responses.time_filter_fast import TimeFilterFastResponse
-from traveltimepy.itertools import split, flatten
+from traveltimepy.itertools import flatten, split
 
 
 class Transportation(BaseModel):