[runpod] Test the minimum number of locations

dstackai · May 8, 2024 · 529a829 · 529a829
1 parent 6cc6c02
commit 529a829
Showing 1 changed file with 5 additions and 7 deletions.
diff --git a/src/integrity_tests/test_runpod.py b/src/integrity_tests/test_runpod.py
@@ -23,20 +23,18 @@ def test_locations(data_rows):
     expected = {
         "CA-MTL-1",
         "CA-MTL-2",
+        "CA-MTL-3",
         "EU-NL-1",
         "EU-RO-1",
         "EU-SE-1",
         "EUR-IS-1",
         "EUR-IS-2",
-        "EUR-NO-1",
+        "US-GA-1",
         "US-OR-1",
+        "US-TX-3",
     }
-    locations = select_row(data_rows, "location")
-    assert set(locations) == expected
-
-    count = Counter(locations)
-    for loc in expected:
-        assert count[loc] > 1
+    locations = set(select_row(data_rows, "location"))
+    assert len(locations) >= len(expected) - 3
 
 
 def test_spot(data_rows):