From 99ac82cc09ec9ac68c4d58b669ecbfa98e1d6d58 Mon Sep 17 00:00:00 2001 From: Xavier Olive Date: Thu, 26 Sep 2024 16:29:33 +0200 Subject: [PATCH] fix ruff --- src/traffic/core/airspace.py | 3 ++- tests/test_flightplan.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/src/traffic/core/airspace.py b/src/traffic/core/airspace.py index a578d8f1..8c83a6e3 100644 --- a/src/traffic/core/airspace.py +++ b/src/traffic/core/airspace.py @@ -2,6 +2,7 @@ import json from functools import cached_property, lru_cache +from itertools import pairwise from pathlib import Path from typing import ( TYPE_CHECKING, @@ -410,7 +411,7 @@ def unary_union_with_alt(polyalt: AirspaceList) -> AirspaceList: simple_union = unary_union([p for p, *_ in polyalt]) return [ExtrudedPolygon(simple_union, float("-inf"), float("inf"))] results: List[ExtrudedPolygon] = [] - for low, up in zip(slices, slices[1:]): + for low, up in pairwise(slices): matched_poly = [ p for (p, low_, up_) in polyalt diff --git a/tests/test_flightplan.py b/tests/test_flightplan.py index b18c1408..338dc998 100644 --- a/tests/test_flightplan.py +++ b/tests/test_flightplan.py @@ -1,3 +1,4 @@ +from itertools import pairwise from typing import List, cast import pandas as pd @@ -180,7 +181,7 @@ def test_flightplan() -> None: assert any(isinstance(p, SpeedLevel) for p in elts) # we can parse everything assert all(p is not None for p in elts) - for cur_, next_ in zip(elts, elts[1:]): + for cur_, next_ in pairwise(elts): # never two consecutive airways assert not isinstance(cur_, Airway) or not isinstance(next_, Airway) # never two consecutive navaids (coordinate points are ok though)