diff options
Diffstat (limited to 'tests/basics/set_remove.py')
| -rw-r--r-- | tests/basics/set_remove.py | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/tests/basics/set_remove.py b/tests/basics/set_remove.py index 208ab137f..5627516c4 100644 --- a/tests/basics/set_remove.py +++ b/tests/basics/set_remove.py @@ -1,3 +1,4 @@ +# basic test s = {1} print(s.remove(1)) print(list(s)) @@ -7,3 +8,26 @@ 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') |
