diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/basics/dict-del.py | 29 | ||||
| -rw-r--r-- | tests/basics/set_remove.py | 24 |
2 files changed, 45 insertions, 8 deletions
diff --git a/tests/basics/dict-del.py b/tests/basics/dict-del.py index 012ed0cae..d908fea42 100644 --- a/tests/basics/dict-del.py +++ b/tests/basics/dict-del.py @@ -1,8 +1,21 @@ -for i in range(100): - d = dict() - for j in range(100): - d[j] = j - del d[i] - for j in range(100): - if j not in d: - print(j, 'not in d') +for n in range(20): + print('testing dict with {} items'.format(n)) + for i in range(n): + # create dict + d = dict() + for j in range(n): + d[str(j)] = j + print(len(d)) + + # delete an item + del d[str(i)] + print(len(d)) + + # check items + for j in range(n): + if str(j) in d: + if j == i: + print(j, 'in d, but it should not be') + else: + if j != i: + print(j, 'not in d, but it should be') 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') |