Fix tests

tortoise · Apr 23, 2024 · 0cccecc · 0cccecc
1 parent 2c51ca6
commit 0cccecc
Showing 1 changed file with 16 additions and 7 deletions.
diff --git a/tests/test_bulk.py b/tests/test_bulk.py
@@ -68,13 +68,22 @@ async def test_bulk_create_mix_specified(self):
         all_ = await UniqueName.all().order_by("id").values("id", "name")
         self.assertEqual(len(all_), 2000)
 
-        self.assertListSortEqual(
-            all_[1000:], [{"id": id_, "name": None} for id_ in range(10000, 11000)], sorted_key="id"
-        )
-        inc = all_[0]["id"]
-        self.assertListSortEqual(
-            all_[:1000], [{"id": val + inc, "name": None} for val in range(1000)], sorted_key="id"
-        )
+        if all_[0]["id"] == 10000:
+            self.assertListSortEqual(
+                all_[:1000], [{"id": id_, "name": None} for id_ in range(10000, 11000)], sorted_key="id"
+            )
+            inc = all_[1000]["id"]
+            self.assertListSortEqual(
+                all_[1000:], [{"id": val + inc, "name": None} for val in range(1000)], sorted_key="id"
+            )
+        else:
+            inc = all_[0]["id"]
+            self.assertListSortEqual(
+                all_[:1000], [{"id": val + inc, "name": None} for val in range(1000)], sorted_key="id"
+            )
+            self.assertListSortEqual(
+                all_[1000:], [{"id": id_, "name": None} for id_ in range(10000, 11000)], sorted_key="id"
+            )
 
     async def test_bulk_create_uuidpk(self):
         await UUIDPkModel.bulk_create([UUIDPkModel() for _ in range(1000)])