micropython/tests/basics/set_binop.py

73 lines
1.8 KiB
Python

# test set binary operations
sets = [set(), {1}, {1, 2}, {1, 2, 3}, {2, 3}, {2, 3, 5}, {5}, {7}]
for s in sets:
for t in sets:
print(sorted(s), '|', sorted(t), '=', sorted(s | t))
print(sorted(s), '^', sorted(t), '=', sorted(s ^ t))
print(sorted(s), '&', sorted(t), '=', sorted(s & t))
print(sorted(s), '-', sorted(t), '=', sorted(s - t))
u = s.copy()
u |= t
print(sorted(s), "|=", sorted(t), '-->', sorted(u))
u = s.copy()
u ^= t
print(sorted(s), "^=", sorted(t), '-->', sorted(u))
u = s.copy()
u &= t
print(sorted(s), "&=", sorted(t), "-->", sorted(u))
u = s.copy()
u -= t
print(sorted(s), "-=", sorted(t), "-->", sorted(u))
print(sorted(s), '==', sorted(t), '=', s == t)
print(sorted(s), '!=', sorted(t), '=', s != t)
print(sorted(s), '>', sorted(t), '=', s > t)
print(sorted(s), '>=', sorted(t), '=', s >= t)
print(sorted(s), '<', sorted(t), '=', s < t)
print(sorted(s), '<=', sorted(t), '=', s <= t)
print(set('abc') == 1)
# make sure inplace operators modify the set
s1 = s2 = set('abc')
s1 |= set('ad')
print(s1 is s2, len(s1))
s1 = s2 = set('abc')
s1 ^= set('ad')
print(s1 is s2, len(s1))
s1 = s2 = set('abc')
s1 &= set('ad')
print(s1 is s2, len(s1))
s1 = s2 = set('abc')
s1 -= set('ad')
print(s1 is s2, len(s1))
# RHS must be a set
try:
print(set('12') >= '1')
except TypeError:
print('TypeError')
# RHS must be a set
try:
print(set('12') <= '123')
except TypeError:
print('TypeError')
# unsupported operator
try:
set('abc') * set('abc')
except TypeError:
print('TypeError')
# unsupported operator with RHS not a set
try:
set('abc') * 2
except TypeError:
print('TypeError')