fix test_disk_format.py

truenas · May 13, 2024 · b43b4b5 · b43b4b5
1 parent 06dab03
commit b43b4b5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/api2/test_disk_format.py b/tests/api2/test_disk_format.py
@@ -65,7 +65,8 @@ def test_disk_format(unused_disk):
     assert partitions[0]['partition_type'] == DATA_TYPE_UUID
 
     # Should be a modulo of grain_size
-    assert (partitions[0]['end_sector'] - partitions[0]['start_sector']) % grain_size == 0
+    assert partitions[0]['size'] % grain_size == 0
+    assert partitions[0]['start_sector'] % grain_size == 0
 
     # Uses (almost) all the disk
     assert partitions[0]['start_sector'] == first_sector