aboutsummaryrefslogtreecommitdiff
path: root/circuitpython/extmod/ulab/tests/1d/numpy/slicing.py
blob: 466c3b2c73767d8421874c12534f0a49e48819df (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
try:
    from ulab import numpy as np
except:
    import numpy as np

for num in range(1,4):
    for start in range(-num, num+1):
        for end in range(-num, num+1):
            for stride in (-3, -2, -1, 1, 2, 3):
                l = list(range(num))
                a = np.array(l, dtype=np.int8)
                sl = l[start:end:stride]
                ll = len(sl)
                try:
                    sa = list(a[start:end:stride])
                except IndexError as e:
                    sa = str(e)
                print("%2d [% d:% d:% d]     %-24r %-24r%s" % (
                    num, start, end, stride, sl, sa, " ***" if sa != sl else ""))

                a[start:end:stride] = np.ones(len(sl)) * -1
                print("%2d [% d:% d:% d]     %r" % (
                    num, start, end, stride, list(a)))