More forgiving size assertion in test_arange

honno · honno · commit 3a90690543b0 · 2021-10-28T10:21:35.000+01:00
diff --git a/array_api_tests/test_creation_functions.py b/array_api_tests/test_creation_functions.py
@@ -228,7 +228,7 @@ def test_arange(dtype, data):
         #     [0.0, 0.33, 0.66, 1.0, 1.33, 1.66]
         #
         min_size = math.floor(size * 0.9)
-        max_size = math.ceil(size * 1.1)
+        max_size = max(math.ceil(size * 1.1), 1)
         assert (
             min_size <= out.size <= max_size
         ), f"{out.size=}, but should be roughly {size} {f_func}"

Original file line number	Diff line number	Diff line change
`@@ -228,7 +228,7 @@ def test_arange(dtype, data):`
`228`	`228`	`# [0.0, 0.33, 0.66, 1.0, 1.33, 1.66]`
`229`	`229`	`#`
`230`	`230`	`min_size = math.floor(size * 0.9)`
`231`		`- max_size = math.ceil(size * 1.1)`
	`231`	`+ max_size = max(math.ceil(size * 1.1), 1)`
`232`	`232`	`assert (`
`233`	`233`	`min_size <= out.size <= max_size`
`234`	`234`	`), f"{out.size=}, but should be roughly {size} {f_func}"`