| #!/usr/bin/env python |
| |
| import unittest, operator, copy, pickle |
| from sets import Set, ImmutableSet |
| from test import test_support |
| |
| empty_set = Set() |
| |
| #============================================================================== |
| |
| class TestBasicOps(unittest.TestCase): |
| |
| def test_repr(self): |
| if self.repr is not None: |
| self.assertEqual(`self.set`, self.repr) |
| |
| def test_length(self): |
| self.assertEqual(len(self.set), self.length) |
| |
| def test_self_equality(self): |
| self.assertEqual(self.set, self.set) |
| |
| def test_equivalent_equality(self): |
| self.assertEqual(self.set, self.dup) |
| |
| def test_copy(self): |
| self.assertEqual(self.set.copy(), self.dup) |
| |
| def test_self_union(self): |
| result = self.set | self.set |
| self.assertEqual(result, self.dup) |
| |
| def test_empty_union(self): |
| result = self.set | empty_set |
| self.assertEqual(result, self.dup) |
| |
| def test_union_empty(self): |
| result = empty_set | self.set |
| self.assertEqual(result, self.dup) |
| |
| def test_self_intersection(self): |
| result = self.set & self.set |
| self.assertEqual(result, self.dup) |
| |
| def test_empty_intersection(self): |
| result = self.set & empty_set |
| self.assertEqual(result, empty_set) |
| |
| def test_intersection_empty(self): |
| result = empty_set & self.set |
| self.assertEqual(result, empty_set) |
| |
| def test_self_symmetric_difference(self): |
| result = self.set ^ self.set |
| self.assertEqual(result, empty_set) |
| |
| def checkempty_symmetric_difference(self): |
| result = self.set ^ empty_set |
| self.assertEqual(result, self.set) |
| |
| def test_self_difference(self): |
| result = self.set - self.set |
| self.assertEqual(result, empty_set) |
| |
| def test_empty_difference(self): |
| result = self.set - empty_set |
| self.assertEqual(result, self.dup) |
| |
| def test_empty_difference_rev(self): |
| result = empty_set - self.set |
| self.assertEqual(result, empty_set) |
| |
| def test_iteration(self): |
| for v in self.set: |
| self.assert_(v in self.values) |
| |
| def test_pickling(self): |
| p = pickle.dumps(self.set) |
| copy = pickle.loads(p) |
| self.assertEqual(self.set, copy, |
| "%s != %s" % (self.set, copy)) |
| |
| #------------------------------------------------------------------------------ |
| |
| class TestBasicOpsEmpty(TestBasicOps): |
| def setUp(self): |
| self.case = "empty set" |
| self.values = [] |
| self.set = Set(self.values) |
| self.dup = Set(self.values) |
| self.length = 0 |
| self.repr = "Set([])" |
| |
| #------------------------------------------------------------------------------ |
| |
| class TestBasicOpsSingleton(TestBasicOps): |
| def setUp(self): |
| self.case = "unit set (number)" |
| self.values = [3] |
| self.set = Set(self.values) |
| self.dup = Set(self.values) |
| self.length = 1 |
| self.repr = "Set([3])" |
| |
| def test_in(self): |
| self.failUnless(3 in self.set) |
| |
| def test_not_in(self): |
| self.failUnless(2 not in self.set) |
| |
| #------------------------------------------------------------------------------ |
| |
| class TestBasicOpsTuple(TestBasicOps): |
| def setUp(self): |
| self.case = "unit set (tuple)" |
| self.values = [(0, "zero")] |
| self.set = Set(self.values) |
| self.dup = Set(self.values) |
| self.length = 1 |
| self.repr = "Set([(0, 'zero')])" |
| |
| def test_in(self): |
| self.failUnless((0, "zero") in self.set) |
| |
| def test_not_in(self): |
| self.failUnless(9 not in self.set) |
| |
| #------------------------------------------------------------------------------ |
| |
| class TestBasicOpsTriple(TestBasicOps): |
| def setUp(self): |
| self.case = "triple set" |
| self.values = [0, "zero", operator.add] |
| self.set = Set(self.values) |
| self.dup = Set(self.values) |
| self.length = 3 |
| self.repr = None |
| |
| #============================================================================== |
| |
| def baditer(): |
| raise TypeError |
| yield True |
| |
| def gooditer(): |
| yield True |
| |
| class TestExceptionPropagation(unittest.TestCase): |
| """SF 628246: Set constructor should not trap iterator TypeErrors""" |
| |
| def test_instanceWithException(self): |
| self.assertRaises(TypeError, Set, baditer()) |
| |
| def test_instancesWithoutException(self): |
| """All of these iterables should load without exception.""" |
| Set([1,2,3]) |
| Set((1,2,3)) |
| Set({'one':1, 'two':2, 'three':3}) |
| Set(xrange(3)) |
| Set('abc') |
| Set(gooditer()) |
| |
| #============================================================================== |
| |
| class TestSetOfSets(unittest.TestCase): |
| def test_constructor(self): |
| inner = Set([1]) |
| outer = Set([inner]) |
| element = outer.pop() |
| self.assertEqual(type(element), ImmutableSet) |
| outer.add(inner) # Rebuild set of sets with .add method |
| outer.remove(inner) |
| self.assertEqual(outer, Set()) # Verify that remove worked |
| outer.discard(inner) # Absence of KeyError indicates working fine |
| |
| #============================================================================== |
| |
| class TestBinaryOps(unittest.TestCase): |
| def setUp(self): |
| self.set = Set((2, 4, 6)) |
| |
| def test_eq(self): # SF bug 643115 |
| self.assertEqual(self.set, Set({2:1,4:3,6:5})) |
| |
| def test_union_subset(self): |
| result = self.set | Set([2]) |
| self.assertEqual(result, Set((2, 4, 6))) |
| |
| def test_union_superset(self): |
| result = self.set | Set([2, 4, 6, 8]) |
| self.assertEqual(result, Set([2, 4, 6, 8])) |
| |
| def test_union_overlap(self): |
| result = self.set | Set([3, 4, 5]) |
| self.assertEqual(result, Set([2, 3, 4, 5, 6])) |
| |
| def test_union_non_overlap(self): |
| result = self.set | Set([8]) |
| self.assertEqual(result, Set([2, 4, 6, 8])) |
| |
| def test_intersection_subset(self): |
| result = self.set & Set((2, 4)) |
| self.assertEqual(result, Set((2, 4))) |
| |
| def test_intersection_superset(self): |
| result = self.set & Set([2, 4, 6, 8]) |
| self.assertEqual(result, Set([2, 4, 6])) |
| |
| def test_intersection_overlap(self): |
| result = self.set & Set([3, 4, 5]) |
| self.assertEqual(result, Set([4])) |
| |
| def test_intersection_non_overlap(self): |
| result = self.set & Set([8]) |
| self.assertEqual(result, empty_set) |
| |
| def test_sym_difference_subset(self): |
| result = self.set ^ Set((2, 4)) |
| self.assertEqual(result, Set([6])) |
| |
| def test_sym_difference_superset(self): |
| result = self.set ^ Set((2, 4, 6, 8)) |
| self.assertEqual(result, Set([8])) |
| |
| def test_sym_difference_overlap(self): |
| result = self.set ^ Set((3, 4, 5)) |
| self.assertEqual(result, Set([2, 3, 5, 6])) |
| |
| def test_sym_difference_non_overlap(self): |
| result = self.set ^ Set([8]) |
| self.assertEqual(result, Set([2, 4, 6, 8])) |
| |
| def test_cmp(self): |
| a, b = Set('a'), Set('b') |
| self.assertRaises(TypeError, cmp, a, b) |
| |
| # You can view this as a buglet: cmp(a, a) does not raise TypeError, |
| # because __eq__ is tried before __cmp__, and a.__eq__(a) returns True, |
| # which Python thinks is good enough to synthesize a cmp() result |
| # without calling __cmp__. |
| self.assertEqual(cmp(a, a), 0) |
| |
| self.assertRaises(TypeError, cmp, a, 12) |
| self.assertRaises(TypeError, cmp, "abc", a) |
| |
| #============================================================================== |
| |
| class TestUpdateOps(unittest.TestCase): |
| def setUp(self): |
| self.set = Set((2, 4, 6)) |
| |
| def test_union_subset(self): |
| self.set |= Set([2]) |
| self.assertEqual(self.set, Set((2, 4, 6))) |
| |
| def test_union_superset(self): |
| self.set |= Set([2, 4, 6, 8]) |
| self.assertEqual(self.set, Set([2, 4, 6, 8])) |
| |
| def test_union_overlap(self): |
| self.set |= Set([3, 4, 5]) |
| self.assertEqual(self.set, Set([2, 3, 4, 5, 6])) |
| |
| def test_union_non_overlap(self): |
| self.set |= Set([8]) |
| self.assertEqual(self.set, Set([2, 4, 6, 8])) |
| |
| def test_union_method_call(self): |
| self.set.union_update(Set([3, 4, 5])) |
| self.assertEqual(self.set, Set([2, 3, 4, 5, 6])) |
| |
| def test_intersection_subset(self): |
| self.set &= Set((2, 4)) |
| self.assertEqual(self.set, Set((2, 4))) |
| |
| def test_intersection_superset(self): |
| self.set &= Set([2, 4, 6, 8]) |
| self.assertEqual(self.set, Set([2, 4, 6])) |
| |
| def test_intersection_overlap(self): |
| self.set &= Set([3, 4, 5]) |
| self.assertEqual(self.set, Set([4])) |
| |
| def test_intersection_non_overlap(self): |
| self.set &= Set([8]) |
| self.assertEqual(self.set, empty_set) |
| |
| def test_intersection_method_call(self): |
| self.set.intersection_update(Set([3, 4, 5])) |
| self.assertEqual(self.set, Set([4])) |
| |
| def test_sym_difference_subset(self): |
| self.set ^= Set((2, 4)) |
| self.assertEqual(self.set, Set([6])) |
| |
| def test_sym_difference_superset(self): |
| self.set ^= Set((2, 4, 6, 8)) |
| self.assertEqual(self.set, Set([8])) |
| |
| def test_sym_difference_overlap(self): |
| self.set ^= Set((3, 4, 5)) |
| self.assertEqual(self.set, Set([2, 3, 5, 6])) |
| |
| def test_sym_difference_non_overlap(self): |
| self.set ^= Set([8]) |
| self.assertEqual(self.set, Set([2, 4, 6, 8])) |
| |
| def test_sym_difference_method_call(self): |
| self.set.symmetric_difference_update(Set([3, 4, 5])) |
| self.assertEqual(self.set, Set([2, 3, 5, 6])) |
| |
| def test_difference_subset(self): |
| self.set -= Set((2, 4)) |
| self.assertEqual(self.set, Set([6])) |
| |
| def test_difference_superset(self): |
| self.set -= Set((2, 4, 6, 8)) |
| self.assertEqual(self.set, Set([])) |
| |
| def test_difference_overlap(self): |
| self.set -= Set((3, 4, 5)) |
| self.assertEqual(self.set, Set([2, 6])) |
| |
| def test_difference_non_overlap(self): |
| self.set -= Set([8]) |
| self.assertEqual(self.set, Set([2, 4, 6])) |
| |
| def test_difference_method_call(self): |
| self.set.difference_update(Set([3, 4, 5])) |
| self.assertEqual(self.set, Set([2, 6])) |
| |
| #============================================================================== |
| |
| class TestMutate(unittest.TestCase): |
| def setUp(self): |
| self.values = ["a", "b", "c"] |
| self.set = Set(self.values) |
| |
| def test_add_present(self): |
| self.set.add("c") |
| self.assertEqual(self.set, Set("abc")) |
| |
| def test_add_absent(self): |
| self.set.add("d") |
| self.assertEqual(self.set, Set("abcd")) |
| |
| def test_add_until_full(self): |
| tmp = Set() |
| expected_len = 0 |
| for v in self.values: |
| tmp.add(v) |
| expected_len += 1 |
| self.assertEqual(len(tmp), expected_len) |
| self.assertEqual(tmp, self.set) |
| |
| def test_remove_present(self): |
| self.set.remove("b") |
| self.assertEqual(self.set, Set("ac")) |
| |
| def test_remove_absent(self): |
| try: |
| self.set.remove("d") |
| self.fail("Removing missing element should have raised LookupError") |
| except LookupError: |
| pass |
| |
| def test_remove_until_empty(self): |
| expected_len = len(self.set) |
| for v in self.values: |
| self.set.remove(v) |
| expected_len -= 1 |
| self.assertEqual(len(self.set), expected_len) |
| |
| def test_discard_present(self): |
| self.set.discard("c") |
| self.assertEqual(self.set, Set("ab")) |
| |
| def test_discard_absent(self): |
| self.set.discard("d") |
| self.assertEqual(self.set, Set("abc")) |
| |
| def test_clear(self): |
| self.set.clear() |
| self.assertEqual(len(self.set), 0) |
| |
| def test_pop(self): |
| popped = {} |
| while self.set: |
| popped[self.set.pop()] = None |
| self.assertEqual(len(popped), len(self.values)) |
| for v in self.values: |
| self.failUnless(v in popped) |
| |
| def test_update_empty_tuple(self): |
| self.set.update(()) |
| self.assertEqual(self.set, Set(self.values)) |
| |
| def test_update_unit_tuple_overlap(self): |
| self.set.update(("a",)) |
| self.assertEqual(self.set, Set(self.values)) |
| |
| def test_update_unit_tuple_non_overlap(self): |
| self.set.update(("a", "z")) |
| self.assertEqual(self.set, Set(self.values + ["z"])) |
| |
| #============================================================================== |
| |
| class TestSubsets(unittest.TestCase): |
| |
| case2method = {"<=": "issubset", |
| ">=": "issuperset", |
| } |
| |
| reverse = {"==": "==", |
| "!=": "!=", |
| "<": ">", |
| ">": "<", |
| "<=": ">=", |
| ">=": "<=", |
| } |
| |
| def test_issubset(self): |
| x = self.left |
| y = self.right |
| for case in "!=", "==", "<", "<=", ">", ">=": |
| expected = case in self.cases |
| # Test the binary infix spelling. |
| result = eval("x" + case + "y", locals()) |
| self.assertEqual(result, expected) |
| # Test the "friendly" method-name spelling, if one exists. |
| if case in TestSubsets.case2method: |
| method = getattr(x, TestSubsets.case2method[case]) |
| result = method(y) |
| self.assertEqual(result, expected) |
| |
| # Now do the same for the operands reversed. |
| rcase = TestSubsets.reverse[case] |
| result = eval("y" + rcase + "x", locals()) |
| self.assertEqual(result, expected) |
| if rcase in TestSubsets.case2method: |
| method = getattr(y, TestSubsets.case2method[rcase]) |
| result = method(x) |
| self.assertEqual(result, expected) |
| #------------------------------------------------------------------------------ |
| |
| class TestSubsetEqualEmpty(TestSubsets): |
| left = Set() |
| right = Set() |
| name = "both empty" |
| cases = "==", "<=", ">=" |
| |
| #------------------------------------------------------------------------------ |
| |
| class TestSubsetEqualNonEmpty(TestSubsets): |
| left = Set([1, 2]) |
| right = Set([1, 2]) |
| name = "equal pair" |
| cases = "==", "<=", ">=" |
| |
| #------------------------------------------------------------------------------ |
| |
| class TestSubsetEmptyNonEmpty(TestSubsets): |
| left = Set() |
| right = Set([1, 2]) |
| name = "one empty, one non-empty" |
| cases = "!=", "<", "<=" |
| |
| #------------------------------------------------------------------------------ |
| |
| class TestSubsetPartial(TestSubsets): |
| left = Set([1]) |
| right = Set([1, 2]) |
| name = "one a non-empty proper subset of other" |
| cases = "!=", "<", "<=" |
| |
| #------------------------------------------------------------------------------ |
| |
| class TestSubsetNonOverlap(TestSubsets): |
| left = Set([1]) |
| right = Set([2]) |
| name = "neither empty, neither contains" |
| cases = "!=" |
| |
| #============================================================================== |
| |
| class TestOnlySetsInBinaryOps(unittest.TestCase): |
| |
| def test_eq_ne(self): |
| # Unlike the others, this is testing that == and != *are* allowed. |
| self.assertEqual(self.other == self.set, False) |
| self.assertEqual(self.set == self.other, False) |
| self.assertEqual(self.other != self.set, True) |
| self.assertEqual(self.set != self.other, True) |
| |
| def test_ge_gt_le_lt(self): |
| self.assertRaises(TypeError, lambda: self.set < self.other) |
| self.assertRaises(TypeError, lambda: self.set <= self.other) |
| self.assertRaises(TypeError, lambda: self.set > self.other) |
| self.assertRaises(TypeError, lambda: self.set >= self.other) |
| |
| self.assertRaises(TypeError, lambda: self.other < self.set) |
| self.assertRaises(TypeError, lambda: self.other <= self.set) |
| self.assertRaises(TypeError, lambda: self.other > self.set) |
| self.assertRaises(TypeError, lambda: self.other >= self.set) |
| |
| def test_union_update(self): |
| try: |
| self.set |= self.other |
| except TypeError: |
| pass |
| else: |
| self.fail("expected TypeError") |
| |
| def test_union(self): |
| self.assertRaises(TypeError, lambda: self.set | self.other) |
| self.assertRaises(TypeError, lambda: self.other | self.set) |
| |
| def test_intersection_update(self): |
| try: |
| self.set &= self.other |
| except TypeError: |
| pass |
| else: |
| self.fail("expected TypeError") |
| |
| def test_intersection(self): |
| self.assertRaises(TypeError, lambda: self.set & self.other) |
| self.assertRaises(TypeError, lambda: self.other & self.set) |
| |
| def test_sym_difference_update(self): |
| try: |
| self.set ^= self.other |
| except TypeError: |
| pass |
| else: |
| self.fail("expected TypeError") |
| |
| def test_sym_difference(self): |
| self.assertRaises(TypeError, lambda: self.set ^ self.other) |
| self.assertRaises(TypeError, lambda: self.other ^ self.set) |
| |
| def test_difference_update(self): |
| try: |
| self.set -= self.other |
| except TypeError: |
| pass |
| else: |
| self.fail("expected TypeError") |
| |
| def test_difference(self): |
| self.assertRaises(TypeError, lambda: self.set - self.other) |
| self.assertRaises(TypeError, lambda: self.other - self.set) |
| |
| #------------------------------------------------------------------------------ |
| |
| class TestOnlySetsNumeric(TestOnlySetsInBinaryOps): |
| def setUp(self): |
| self.set = Set((1, 2, 3)) |
| self.other = 19 |
| |
| #------------------------------------------------------------------------------ |
| |
| class TestOnlySetsDict(TestOnlySetsInBinaryOps): |
| def setUp(self): |
| self.set = Set((1, 2, 3)) |
| self.other = {1:2, 3:4} |
| |
| #------------------------------------------------------------------------------ |
| |
| class TestOnlySetsOperator(TestOnlySetsInBinaryOps): |
| def setUp(self): |
| self.set = Set((1, 2, 3)) |
| self.other = operator.add |
| |
| #============================================================================== |
| |
| class TestCopying(unittest.TestCase): |
| |
| def test_copy(self): |
| dup = self.set.copy() |
| dup_list = list(dup); dup_list.sort() |
| set_list = list(self.set); set_list.sort() |
| self.assertEqual(len(dup_list), len(set_list)) |
| for i in range(len(dup_list)): |
| self.failUnless(dup_list[i] is set_list[i]) |
| |
| def test_deep_copy(self): |
| dup = copy.deepcopy(self.set) |
| ##print type(dup), `dup` |
| dup_list = list(dup); dup_list.sort() |
| set_list = list(self.set); set_list.sort() |
| self.assertEqual(len(dup_list), len(set_list)) |
| for i in range(len(dup_list)): |
| self.assertEqual(dup_list[i], set_list[i]) |
| |
| #------------------------------------------------------------------------------ |
| |
| class TestCopyingEmpty(TestCopying): |
| def setUp(self): |
| self.set = Set() |
| |
| #------------------------------------------------------------------------------ |
| |
| class TestCopyingSingleton(TestCopying): |
| def setUp(self): |
| self.set = Set(["hello"]) |
| |
| #------------------------------------------------------------------------------ |
| |
| class TestCopyingTriple(TestCopying): |
| def setUp(self): |
| self.set = Set(["zero", 0, None]) |
| |
| #------------------------------------------------------------------------------ |
| |
| class TestCopyingTuple(TestCopying): |
| def setUp(self): |
| self.set = Set([(1, 2)]) |
| |
| #------------------------------------------------------------------------------ |
| |
| class TestCopyingNested(TestCopying): |
| def setUp(self): |
| self.set = Set([((1, 2), (3, 4))]) |
| |
| #============================================================================== |
| |
| libreftest = """ |
| Example from the Library Reference: Doc/lib/libsets.tex |
| |
| >>> from sets import Set as Base # override _repr to get sorted output |
| >>> class Set(Base): |
| ... def _repr(self): |
| ... return Base._repr(self, sorted=True) |
| >>> engineers = Set(['John', 'Jane', 'Jack', 'Janice']) |
| >>> programmers = Set(['Jack', 'Sam', 'Susan', 'Janice']) |
| >>> management = Set(['Jane', 'Jack', 'Susan', 'Zack']) |
| >>> employees = engineers | programmers | management # union |
| >>> engineering_management = engineers & programmers # intersection |
| >>> fulltime_management = management - engineers - programmers # difference |
| >>> engineers.add('Marvin') |
| >>> print engineers |
| Set(['Jack', 'Jane', 'Janice', 'John', 'Marvin']) |
| >>> employees.issuperset(engineers) # superset test |
| False |
| >>> employees.update(engineers) # update from another set |
| >>> employees.issuperset(engineers) |
| True |
| >>> for group in [engineers, programmers, management, employees]: |
| ... group.discard('Susan') # unconditionally remove element |
| ... print group |
| ... |
| Set(['Jack', 'Jane', 'Janice', 'John', 'Marvin']) |
| Set(['Jack', 'Janice', 'Sam']) |
| Set(['Jack', 'Jane', 'Zack']) |
| Set(['Jack', 'Jane', 'Janice', 'John', 'Marvin', 'Sam', 'Zack']) |
| """ |
| |
| #============================================================================== |
| |
| __test__ = {'libreftest' : libreftest} |
| |
| def test_main(verbose=None): |
| from test import test_sets |
| test_support.run_unittest( |
| TestSetOfSets, |
| TestExceptionPropagation, |
| TestBasicOpsEmpty, |
| TestBasicOpsSingleton, |
| TestBasicOpsTuple, |
| TestBasicOpsTriple, |
| TestBinaryOps, |
| TestUpdateOps, |
| TestMutate, |
| TestSubsetEqualEmpty, |
| TestSubsetEqualNonEmpty, |
| TestSubsetEmptyNonEmpty, |
| TestSubsetPartial, |
| TestSubsetNonOverlap, |
| TestOnlySetsNumeric, |
| TestOnlySetsDict, |
| TestOnlySetsOperator, |
| TestCopyingEmpty, |
| TestCopyingSingleton, |
| TestCopyingTriple, |
| TestCopyingTuple, |
| TestCopyingNested |
| ) |
| test_support.run_doctest(test_sets, verbose) |
| |
| if __name__ == "__main__": |
| test_main(verbose=True) |