Hau-coder
diff --git a/‎tests/snippets/set.py
Lines changed: 85 additions & 0 deletions b/‎tests/snippets/set.py
Lines changed: 85 additions & 0 deletions
@@ -144,3 +144,88 @@ def __hash__(self):
 assert a == set([1,2,4,5])
 with assertRaises(TypeError):
 	a ^= 1
+
+# frozen set
+
+assert frozenset([1,2]) == frozenset([1,2])
+assert not frozenset([1,2,3]) == frozenset([1,2])
+
+assert frozenset([1,2,3]) >= frozenset([1,2])
+assert frozenset([1,2]) >= frozenset([1,2])
+assert not frozenset([1,3]) >= frozenset([1,2])
+
+assert frozenset([1,2,3]).issuperset(frozenset([1,2]))
+assert frozenset([1,2]).issuperset(frozenset([1,2]))
+assert not frozenset([1,3]).issuperset(frozenset([1,2]))
+
+assert frozenset([1,2,3]) > frozenset([1,2])
+assert not frozenset([1,2]) > frozenset([1,2])
+assert not frozenset([1,3]) > frozenset([1,2])
+
+assert frozenset([1,2]) <= frozenset([1,2,3])
+assert frozenset([1,2]) <= frozenset([1,2])
+assert not frozenset([1,3]) <= frozenset([1,2])
+
+assert frozenset([1,2]).issubset(frozenset([1,2,3]))
+assert frozenset([1,2]).issubset(frozenset([1,2]))
+assert not frozenset([1,3]).issubset(frozenset([1,2]))
+
+assert frozenset([1,2]) < frozenset([1,2,3])
+assert not frozenset([1,2]) < frozenset([1,2])
+assert not frozenset([1,3]) < frozenset([1,2])
+
+a = frozenset([1, 2, 3])
+assert len(a) == 3
+b = a.copy()
+assert b == a
+
+assert frozenset([1,2,3]).union(frozenset([4,5])) == frozenset([1,2,3,4,5])
+assert frozenset([1,2,3]).union(frozenset([1,2,3,4,5])) == frozenset([1,2,3,4,5])
+
+assert frozenset([1,2,3]) | frozenset([4,5]) == frozenset([1,2,3,4,5])
+assert frozenset([1,2,3]) | frozenset([1,2,3,4,5]) == frozenset([1,2,3,4,5])
+
+assert frozenset([1,2,3]).intersection(frozenset([1,2])) == frozenset([1,2])
+assert frozenset([1,2,3]).intersection(frozenset([5,6])) == frozenset([])
+
+assert frozenset([1,2,3]) & frozenset([4,5]) == frozenset([])
+assert frozenset([1,2,3]) & frozenset([1,2,3,4,5]) == frozenset([1,2,3])
+
+assert frozenset([1,2,3]).difference(frozenset([1,2])) == frozenset([3])
+assert frozenset([1,2,3]).difference(frozenset([5,6])) == frozenset([1,2,3])
+
+assert frozenset([1,2,3]) - frozenset([4,5]) == frozenset([1,2,3])
+assert frozenset([1,2,3]) - frozenset([1,2,3,4,5]) == frozenset([])
+
+assert frozenset([1,2,3]).symmetric_difference(frozenset([1,2])) == frozenset([3])
+assert frozenset([1,2,3]).symmetric_difference(frozenset([5,6])) == frozenset([1,2,3,5,6])
+
+assert frozenset([1,2,3]) ^ frozenset([4,5]) == frozenset([1,2,3,4,5])
+assert frozenset([1,2,3]) ^ frozenset([1,2,3,4,5]) == frozenset([4,5])
+
+assert_raises(TypeError, lambda: frozenset([[]]))
+
+# set and frozen set
+assert frozenset([1,2,3]).union(set([4,5])) == frozenset([1,2,3,4,5])
+assert set([1,2,3]).union(frozenset([4,5])) == set([1,2,3,4,5])
+
+assert frozenset([1,2,3]) | set([4,5]) == frozenset([1,2,3,4,5])
+assert set([1,2,3]) | frozenset([4,5]) == set([1,2,3,4,5])
+
+assert frozenset([1,2,3]).intersection(set([5,6])) == frozenset([])
+assert set([1,2,3]).intersection(frozenset([5,6])) == set([])
+
+assert frozenset([1,2,3]) & set([1,2,3,4,5]) == frozenset([1,2,3])
+assert set([1,2,3]) & frozenset([1,2,3,4,5]) == set([1,2,3])
+
+assert frozenset([1,2,3]).difference(set([5,6])) == frozenset([1,2,3])
+assert set([1,2,3]).difference(frozenset([5,6])) == set([1,2,3])
+
+assert frozenset([1,2,3]) - set([4,5]) == frozenset([1,2,3])
+assert set([1,2,3]) - frozenset([4,5]) == frozenset([1,2,3])
+
+assert frozenset([1,2,3]).symmetric_difference(set([1,2])) == frozenset([3])
+assert set([1,2,3]).symmetric_difference(frozenset([1,2])) == set([3])
+
+assert frozenset([1,2,3]) ^ set([4,5]) == frozenset([1,2,3,4,5])
+assert set([1,2,3]) ^ frozenset([4,5]) == set([1,2,3,4,5])