Skip to content

Commit 7984012

Browse files
committed
Add test cases for __sub__ and __rsub__ methods of set and frozenset
1 parent 05d0248 commit 7984012

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

extra_tests/snippets/builtin_set.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -103,6 +103,9 @@ class S(set):
103103
assert set([1,2,3]) - set([1,2,3,4,5]) == set([])
104104
assert_raises(TypeError, lambda: set([1,2,3]) - [1,2,3,4,5])
105105

106+
assert set([1,2]).__sub__(set([2,3])) == set([1])
107+
assert set([1,2]).__rsub__(set([2,3])) == set([3])
108+
106109
assert set([1,2,3]).symmetric_difference(set([1,2])) == set([3])
107110
assert set([1,2,3]).symmetric_difference(set([5,6])) == set([1,2,3,5,6])
108111
assert set([1,2,3]).symmetric_difference([1,2]) == set([3])
@@ -271,6 +274,9 @@ class S(set):
271274
assert frozenset([1,2,3]) - frozenset([1,2,3,4,5]) == frozenset([])
272275
assert_raises(TypeError, lambda: frozenset([1,2,3]) - [1,2,3,4,5])
273276

277+
assert frozenset([1,2]).__sub__(frozenset([2,3])) == frozenset([1])
278+
assert frozenset([1,2]).__rsub__(frozenset([2,3])) == frozenset([3])
279+
274280
assert frozenset([1,2,3]).symmetric_difference(frozenset([1,2])) == frozenset([3])
275281
assert frozenset([1,2,3]).symmetric_difference(frozenset([5,6])) == frozenset([1,2,3,5,6])
276282
assert frozenset([1,2,3]).symmetric_difference([1,2]) == frozenset([3])
@@ -311,6 +317,11 @@ class S(set):
311317
assert frozenset([1,2,3]) - set([4,5]) == frozenset([1,2,3])
312318
assert set([1,2,3]) - frozenset([4,5]) == frozenset([1,2,3])
313319

320+
assert frozenset([1,2]).__sub__(set([2,3])) == frozenset([1])
321+
assert frozenset([1,2]).__rsub__(set([2,3])) == set([3])
322+
assert set([1,2]).__sub__(frozenset([2,3])) == set([1])
323+
assert set([1,2]).__rsub__(frozenset([2,3])) == frozenset([3])
324+
314325
assert frozenset([1,2,3]).symmetric_difference(set([1,2])) == frozenset([3])
315326
assert set([1,2,3]).symmetric_difference(frozenset([1,2])) == set([3])
316327

0 commit comments

Comments
 (0)