aboutsummaryrefslogtreecommitdiff
path: root/circuitpython/extmod/ulab/tests/2d/numpy/operators.py
blob: acc316b16c65b38e0403c09b5e721606db49fd81 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
try:
    from ulab import numpy as np
except ImportError:
    import numpy as np

  
print(len(np.array([1, 2, 3, 4, 5], dtype=np.uint8)))
print(len(np.array([[1, 2, 3],[4, 5, 6]])))

print(~np.array([0, -1, -100], dtype=np.uint8))
print(~np.array([0, -1, -100], dtype=np.uint16))
print(~np.array([0, -1, -100], dtype=np.int8))
print(~np.array([0, -1, -100], dtype=np.int16))

print(abs(np.array([0, -1, -100], dtype=np.uint8)))
print(abs(np.array([0, -1, -100], dtype=np.uint16)))
print(abs(np.array([0, -1, -100], dtype=np.int8)))
print(abs(np.array([0, -1, -100], dtype=np.int16)))
print(abs(np.array([0, -1, -100], dtype=np.float)))

print(-(np.array([0, -1, -100], dtype=np.uint8)))
print(-(np.array([0, -1, -100], dtype=np.uint16)))
print(-(np.array([0, -1, -100], dtype=np.int8)))
print(-(np.array([0, -1, -100], dtype=np.int16)))
print(-(np.array([0, -1, -100], dtype=np.float)))

print(+(np.array([0, -1, -100], dtype=np.uint8)))
print(+(np.array([0, -1, -100], dtype=np.uint16)))
print(+(np.array([0, -1, -100], dtype=np.int8)))
print(+(np.array([0, -1, -100], dtype=np.int16)))
print(+(np.array([0, -1, -100], dtype=np.float)))

print(np.array([1,2,3], dtype=np.float) > np.array([4,5,6], dtype=np.float))
print(np.array([1,2,3], dtype=np.float) > np.array([4,5,6], dtype=np.uint16))
print(np.array([1,2,3], dtype=np.float) > np.array([4,5,6], dtype=np.int16))
print(np.array([1,2,3], dtype=np.float) < np.array([4,5,6], dtype=np.float))
print(np.array([1,2,3], dtype=np.float) < np.array([4,5,6], dtype=np.uint16))
print(np.array([1,2,3], dtype=np.float) < np.array([4,5,6], dtype=np.int16))

print(np.array([1,2,3], dtype=np.float) >= np.array([4,5,6], dtype=np.float))
print(np.array([1,2,3], dtype=np.float) >= np.array([4,5,6], dtype=np.uint16))
print(np.array([1,2,3], dtype=np.float) >= np.array([4,5,6], dtype=np.int16))
print(np.array([1,2,3], dtype=np.float) <= np.array([4,5,6], dtype=np.float))
print(np.array([1,2,3], dtype=np.float) <= np.array([4,5,6], dtype=np.uint16))
print(np.array([1,2,3], dtype=np.float) <= np.array([4,5,6], dtype=np.int16))

print(np.array([1,2,3], dtype=np.float) > 4)
print(np.array([1,2,3], dtype=np.float) > 4.0)
print(np.array([1,2,3], dtype=np.float) < 4)
print(np.array([1,2,3], dtype=np.float) < 4.0)

print(np.array([1,2,3], dtype=np.float) == np.array([4,5,6], dtype=np.float))
print(np.array([1,2,3], dtype=np.float) == np.array([4,5,6], dtype=np.uint16))
print(np.array([1,2,3], dtype=np.float) == np.array([4,5,6], dtype=np.int16))
print(np.array([1,2,3], dtype=np.float) != np.array([4,5,6], dtype=np.float))
print(np.array([1,2,3], dtype=np.float) != np.array([4,5,6], dtype=np.uint16))
print(np.array([1,2,3], dtype=np.float) != np.array([4,5,6], dtype=np.int16))

print(np.array([1,2,3], dtype=np.float) == 4)
print(np.array([1,2,3], dtype=np.float) == 4.0)
print(np.array([1,2,3], dtype=np.float) != 4)
print(np.array([1,2,3], dtype=np.float) != 4.0)

print(np.array([1,2,3], dtype=np.float) - np.array([4,5,6], dtype=np.float))
print(np.array([1,2,3], dtype=np.float) - np.array([4,5,6], dtype=np.uint16))
print(np.array([1,2,3], dtype=np.float) - np.array([4,5,6], dtype=np.int16))

print(np.array([1,2,3], dtype=np.float) + np.array([4,5,6], dtype=np.float))
print(np.array([1,2,3], dtype=np.float) + np.array([4,5,6], dtype=np.uint16))
print(np.array([1,2,3], dtype=np.float) + np.array([4,5,6], dtype=np.int16))

print(np.array([1,2,3], dtype=np.float) * np.array([4,5,6], dtype=np.float))
print(np.array([1,2,3], dtype=np.float) * np.array([4,5,6], dtype=np.uint16))
print(np.array([1,2,3], dtype=np.float) * np.array([4,5,6], dtype=np.int16))

print(np.array([1,2,3], dtype=np.float) ** np.array([4,5,6], dtype=np.float))
print(np.array([1,2,3], dtype=np.float) ** np.array([4,5,6], dtype=np.uint16))
print(np.array([1,2,3], dtype=np.float) ** np.array([4,5,6], dtype=np.int16))

print(np.array([1,2,3], dtype=np.float) / np.array([4,5,6], dtype=np.float))
print(np.array([1,2,3], dtype=np.float) / np.array([4,5,6], dtype=np.uint16))
print(np.array([1,2,3], dtype=np.float) / np.array([4,5,6], dtype=np.int16))

print(np.array([1,2,3], dtype=np.float) - 4)
print(np.array([1,2,3], dtype=np.float) - 4.0)
print(np.array([1,2,3], dtype=np.float) + 4)
print(np.array([1,2,3], dtype=np.float) + 4.0)

print(np.array([1,2,3], dtype=np.float) * 4)
print(np.array([1,2,3], dtype=np.float) * 4.0)
print(np.array([1,2,3], dtype=np.float) ** 4)
print(np.array([1,2,3], dtype=np.float) ** 4.0)

print(np.array([1,2,3], dtype=np.float) / 4)
print(np.array([1,2,3], dtype=np.float) / 4.0)

a = np.array([1,2,3], dtype=np.float)
a -= np.array([4,5,6], dtype=np.float)
print(a)

a = np.array([1,2,3], dtype=np.float)
a -= np.array([4,5,6], dtype=np.uint16)
print(a)

a = np.array([1,2,3], dtype=np.float)
a -= np.array([4,5,6], dtype=np.int16)
print(a)

a = np.array([1,2,3], dtype=np.float)
a += np.array([4,5,6], dtype=np.float)
print(a)

a = np.array([1,2,3], dtype=np.float)
a += np.array([4,5,6], dtype=np.uint16)
print(a)

a = np.array([1,2,3], dtype=np.float)
a += np.array([4,5,6], dtype=np.int16)
print(a)

a = np.array([1,2,3], dtype=np.float)
a *= np.array([4,5,6], dtype=np.float)
print(a)

a = np.array([1,2,3], dtype=np.float)
a *= np.array([4,5,6], dtype=np.uint16)
print(a)

a = np.array([1,2,3], dtype=np.float)
a *= np.array([4,5,6], dtype=np.int16)
print(a)

a = np.array([1,2,3], dtype=np.float)
#a /= np.array([4,5,6])
print(a)

a = np.array([1,2,3], dtype=np.float)
a **= np.array([4,5,6], dtype=np.float)
print(a)

a = np.array([1,2,3], dtype=np.float)
a **= np.array([4,5,6], dtype=np.uint16)
print(a)

a = np.array([1,2,3], dtype=np.float)
a **= np.array([4,5,6], dtype=np.int16)
print(a)

print(np.array([1,2,3],dtype=np.uint8) + np.array([4,5,6],dtype=np.uint8))
print(np.array([1,2,3],dtype=np.uint8) + np.array([4,5,6],dtype=np.int8))
print(np.array([1,2,3],dtype=np.int8) + np.array([4,5,6],dtype=np.int8))
print(np.array([1,2,3],dtype=np.uint8) + np.array([4,5,6],dtype=np.uint16))
print(np.array([1,2,3],dtype=np.int8) + np.array([4,5,6],dtype=np.uint16))
print(np.array([1,2,3],dtype=np.uint8) + np.array([4,5,6],dtype=np.int16))
print(np.array([1,2,3],dtype=np.int8) + np.array([4,5,6],dtype=np.int16))
print(np.array([1,2,3],dtype=np.uint16) + np.array([4,5,6],dtype=np.uint16))
print(np.array([1,2,3],dtype=np.int16) + np.array([4,5,6],dtype=np.int16))
print(np.array([1,2,3],dtype=np.int16) + np.array([4,5,6],dtype=np.uint16))

print(np.array([1,2,3],dtype=np.uint8) + np.array([4,5,6],dtype=np.float))
print(np.array([1,2,3],dtype=np.int8) + np.array([4,5,6],dtype=np.float))
print(np.array([1,2,3],dtype=np.uint16) + np.array([4,5,6],dtype=np.float))
print(np.array([1,2,3],dtype=np.int16) + np.array([4,5,6],dtype=np.float))
print(np.array([1,2,3],dtype=np.int16) + np.array([4,5,6],dtype=np.float))

a = np.array([1, 2, 3, 4, 5], dtype=np.uint8)
for i, _a in enumerate(a):
    print("element %d in a:"%i, _a)