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') | 
