add type annotations for set

kiwix · Aug 23, 2024 · ae7183c · ae7183c
1 parent bfd4417
commit ae7183c
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/backend/tests/db/test_mirrors.py b/backend/tests/db/test_mirrors.py
@@ -127,17 +127,14 @@ def test_update_mirror_countries_from_regions(
 ):
 
     regions = [asia_region, africa_region, europe_region]
-    expected_country_codes = set()
-    region_codes = set()
+    expected_country_codes: set[str] = set()
+    region_codes: set[str] = set()
 
     for region in regions:
         region_codes.add(region.code)
         for country in region.countries:
             expected_country_codes.add(country.code)
 
-    if db_mirror.country:
-        expected_country_codes.add(db_mirror.country_code)
-
     db_mirror = update_mirror_countries_from_regions(dbsession, db_mirror, region_codes)
 
     assert db_mirror.other_countries is not None