tests/basics/set_remove.py


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

# basic test
s = {1}
print(s.remove(1))
print(list(s))
try:
    print(s.remove(1), "!!!")
except KeyError:
    pass
else:
    print("failed to raise KeyError")

# test sets of varying size
for n in range(20):
    print('testing set with {} items'.format(n))
    for i in range(n):
        # create set
        s = set()
        for j in range(n):
            s.add(str(j))
        print(len(s))

        # delete an item
        s.remove(str(i))
        print(len(s))

        # check items
        for j in range(n):
            if str(j) in s:
                if j == i:
                    print(j, 'in s, but it should not be')
            else:
                if j != i:
                    print(j, 'not in s, but it should be')