Skip to content

Commit f3f847e

Browse files
committed
Add test
1 parent 64a1465 commit f3f847e

File tree

1 file changed

+20
-0
lines changed

1 file changed

+20
-0
lines changed

test/indexing.jl

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,20 @@
1+
using Test
2+
using oneAPI
3+
4+
@testset "findall" begin
5+
bools1d = oneArray([true, false, true, false, true])
6+
@test Array(findall(bools1d)) == findall(Bool[true, false, true, false, true])
7+
8+
bools2d = oneArray(Bool[true false; false true; true false])
9+
@test Array(findall(bools2d)) == findall(Bool[true false; false true; true false])
10+
11+
all_false = oneArray(fill(false, 4))
12+
@test Array(findall(all_false)) == Int[]
13+
14+
all_true = oneArray(fill(true, 3, 2))
15+
@test Array(findall(all_true)) == findall(fill(true, 3, 2))
16+
17+
data = oneArray(collect(1:6))
18+
mask = oneArray(Bool[true, false, true, false, false, true])
19+
@test Array(data[mask]) == collect(1:6)[findall(Bool[true, false, true, false, false, true])]
20+
end

0 commit comments

Comments
 (0)