Skip to content

Commit 5acbfe1

Browse files
committed
TEST: Added simple test for algorithms
- Makes sure function everything runs
1 parent 54812f8 commit 5acbfe1

File tree

1 file changed

+61
-0
lines changed

1 file changed

+61
-0
lines changed

tests/simple_algorithm.py

Lines changed: 61 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,61 @@
1+
#!/usr/bin/python
2+
import arrayfire as af
3+
4+
a = af.randu(3, 3)
5+
6+
print(af.sum(a), af.product(a), af.min(a), af.max(a), af.count(a), af.any_true(a), af.all_true(a))
7+
8+
af.print_array(af.sum(a, 0))
9+
af.print_array(af.sum(a, 1))
10+
11+
af.print_array(af.product(a, 0))
12+
af.print_array(af.product(a, 1))
13+
14+
af.print_array(af.min(a, 0))
15+
af.print_array(af.min(a, 1))
16+
17+
af.print_array(af.max(a, 0))
18+
af.print_array(af.max(a, 1))
19+
20+
af.print_array(af.count(a, 0))
21+
af.print_array(af.count(a, 1))
22+
23+
af.print_array(af.any_true(a, 0))
24+
af.print_array(af.any_true(a, 1))
25+
26+
af.print_array(af.all_true(a, 0))
27+
af.print_array(af.all_true(a, 1))
28+
29+
af.print_array(af.accum(a, 0))
30+
af.print_array(af.accum(a, 1))
31+
32+
af.print_array(af.sort(a, is_ascending=True))
33+
af.print_array(af.sort(a, is_ascending=False))
34+
35+
val,idx = af.sort_index(a, is_ascending=True)
36+
af.print_array(val)
37+
af.print_array(idx)
38+
val,idx = af.sort_index(a, is_ascending=False)
39+
af.print_array(val)
40+
af.print_array(idx)
41+
42+
b = af.randu(3,3)
43+
keys,vals = af.sort_by_key(a, b, is_ascending=True)
44+
af.print_array(keys)
45+
af.print_array(vals)
46+
keys,vals = af.sort_by_key(a, b, is_ascending=False)
47+
af.print_array(keys)
48+
af.print_array(vals)
49+
50+
c = af.randu(5,1)
51+
d = af.randu(5,1)
52+
cc = af.set_unique(c, is_sorted=False)
53+
dd = af.set_unique(af.sort(d), is_sorted=True)
54+
af.print_array(cc)
55+
af.print_array(dd)
56+
57+
af.print_array(af.set_union(cc, dd, is_unique=True))
58+
af.print_array(af.set_union(cc, dd, is_unique=False))
59+
60+
af.print_array(af.set_intersect(cc, cc, is_unique=True))
61+
af.print_array(af.set_intersect(cc, cc, is_unique=False))

0 commit comments

Comments
 (0)