FilesExpand file tree

 master

/

heapalloc_iter.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

70 lines (62 loc) · 1.3 KB

 master

/

heapalloc_iter.py

Copy path

Top

File metadata and controls

• Code
• Blame

70 lines (62 loc) · 1.3 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

# test that iterating doesn't use the heap

try:

frozenset

import array

except (NameError, ImportError):

print("SKIP")

raise SystemExit

try:

from micropython import heap_lock, heap_unlock

except (ImportError, AttributeError):

heap_lock = heap_unlock = lambda:0

def do_iter(l):

heap_lock()

for i in l:

print(i)

heap_unlock()

def gen_func():

yield 1

yield 2

# pre-create collections to iterate over

ba = bytearray(b'123')

ar = array.array('H', (123, 456))

t = (1, 2, 3)

l = [1, 2]

d = {1:2}

s = set((1,))

fs = frozenset((1,))

g1 = (100 + x for x in range(2))

g2 = gen_func()

# test containment (both success and failure) with the heap locked

heap_lock()

print(49 in b'123', 255 in b'123')

print(1 in t, -1 in t)

print(1 in l, -1 in l)

print(1 in d, -1 in d)

print(1 in s, -1 in s)

heap_unlock()

# test unpacking with the heap locked

unp0 = unp1 = unp2 = None # preallocate slots for globals

heap_lock()

unp0, unp1, unp2 = t

print(unp0, unp1, unp2)

heap_unlock()

# test certain builtins with the heap locked

heap_lock()

print(all(t))

print(any(t))

print(min(t))

print(max(t))

print(sum(t))

heap_unlock()

# test iterating over collections with the heap locked

do_iter(b'123')

do_iter(ba)

do_iter(ar)

do_iter(t)

do_iter(l)

do_iter(d)

do_iter(s)

do_iter(fs)

do_iter(g1)

do_iter(g2)