We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
where
1 parent bde69fd commit 709f4feCopy full SHA for 709f4fe
1 file changed
array_api_strict/tests/test_searching_functions.py
@@ -35,3 +35,10 @@ def test_where_mixed_dtypes():
35
c = x > 1.5
36
res = xp.where(c, False, c)
37
assert all(res == xp.asarray([False, False]))
38
+
39
40
+def test_where_f32():
41
+ # https://github.com/data-apis/array-api-strict/issues/131#issuecomment-2723016294
42
+ res = xp.where(xp.asarray([True, False]), 1., xp.asarray([2, 2], dtype=xp.float32))
43
+ assert res.dtype == xp.float32
44
0 commit comments