improved test

JeanExtreme002 · Dec 21, 2023 · 28711d1 · 28711d1
1 parent 273600b
commit 28711d1
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/python/tests/test_api.py b/python/tests/test_api.py
@@ -87,9 +87,11 @@ def test_get_flights_by_airline(airlines = ["SWA", "GLO", "AZU", "UAL", "THY"],
 
 
 @repeat_test(**repeat_test_config)
-def test_get_flights_by_bounds(zones = ["northamerica", "southamerica"], expect = 30):
-    for zone in zones:
-        zone = fr_api.get_zones()[zone]
+def test_get_flights_by_bounds(target_zones = ["northamerica", "southamerica"], expect = 30):
+    zones = fr_api.get_zones()
+
+    for zone in target_zones:
+        zone = zones[zone]
         bounds = fr_api.get_bounds(zone)
 
         flights = fr_api.get_flights(bounds = bounds)