Skip to content

Commit 503b49d

Browse files
committed
Adding simple tests for operators in af.array class
1 parent be27746 commit 503b49d

File tree

1 file changed

+115
-0
lines changed

1 file changed

+115
-0
lines changed

tests/simple_arith.py

Lines changed: 115 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,115 @@
1+
#!/usr/bin/python
2+
import arrayfire as af
3+
4+
a = af.randu(3,3,dtype=af.u32)
5+
b = af.constant(4, 3, 3, dtype=af.u32)
6+
af.print_array(a)
7+
af.print_array(b)
8+
9+
c = a + b
10+
d = a
11+
d += b
12+
13+
af.print_array(c)
14+
af.print_array(d)
15+
af.print_array(a + 2)
16+
af.print_array(3 + a)
17+
18+
19+
c = a - b
20+
d = a
21+
d -= b
22+
23+
af.print_array(c)
24+
af.print_array(d)
25+
af.print_array(a - 2)
26+
af.print_array(3 - a)
27+
28+
c = a * b
29+
d = a
30+
d *= b
31+
32+
af.print_array(c * 2)
33+
af.print_array(3 * d)
34+
af.print_array(a * 2)
35+
af.print_array(3 * a)
36+
37+
c = a / b
38+
d = a
39+
d /= b
40+
41+
af.print_array(c / 2.0)
42+
af.print_array(3.0 / d)
43+
af.print_array(a / 2)
44+
af.print_array(3 / a)
45+
46+
c = a % b
47+
d = a
48+
d %= b
49+
50+
af.print_array(c % 2.0)
51+
af.print_array(3.0 % d)
52+
af.print_array(a % 2)
53+
af.print_array(3 % a)
54+
55+
c = a ** b
56+
d = a
57+
d **= b
58+
59+
af.print_array(c ** 2.0)
60+
af.print_array(3.0 ** d)
61+
af.print_array(a ** 2)
62+
af.print_array(3 ** a)
63+
64+
af.print_array(a < b)
65+
af.print_array(a < 0.5)
66+
af.print_array(0.5 < a)
67+
68+
af.print_array(a <= b)
69+
af.print_array(a <= 0.5)
70+
af.print_array(0.5 <= a)
71+
72+
af.print_array(a > b)
73+
af.print_array(a > 0.5)
74+
af.print_array(0.5 > a)
75+
76+
af.print_array(a >= b)
77+
af.print_array(a >= 0.5)
78+
af.print_array(0.5 >= a)
79+
80+
af.print_array(a != b)
81+
af.print_array(a != 0.5)
82+
af.print_array(0.5 != a)
83+
84+
af.print_array(a == b)
85+
af.print_array(a == 0.5)
86+
af.print_array(0.5 == a)
87+
88+
af.print_array(a & b)
89+
af.print_array(a & 2)
90+
c = a
91+
c &= 2
92+
af.print_array(c)
93+
94+
af.print_array(a | b)
95+
af.print_array(a | 2)
96+
c = a
97+
c |= 2
98+
af.print_array(c)
99+
100+
af.print_array(a >> b)
101+
af.print_array(a >> 2)
102+
c = a
103+
c >>= 2
104+
af.print_array(c)
105+
106+
af.print_array(a << b)
107+
af.print_array(a << 2)
108+
c = a
109+
c <<= 2
110+
af.print_array(c)
111+
112+
af.print_array(-a)
113+
af.print_array(+a)
114+
af.print_array(~a)
115+
af.print_array(a)

0 commit comments

Comments
 (0)