""" Tests if NSSet conforms to the interface of the python type set() This is a port of the set tests from the Python stdlib for 3.2 """ from PyObjCTools.TestSupport import * import objc import operator NSSet = objc.lookUpClass('NSSet') NSMutableSet = objc.lookUpClass('NSMutableSet') import test.test_set from test.test_set import PassThru, check_pass_thru test.test_set.empty_set = NSMutableSet() class TestSet (test.test_set.TestJointOps, TestCase): thetype = NSSet basetype = NSSet def test_cyclical_print(self): pass def test_cyclical_repr(self): pass def test_pickling(self): pass def test_do_not_rehash_dict_keys(self): pass def test_badcmp(self): pass def test_subclass_with_custom_hash(self): pass def test_deepcopy(self): # XXX: is it possible to get this to work? pass def test_union(self): # Same test as inherrited in python 2.7, but without # type equality tests u = self.s.union(self.otherword) for c in self.letters: self.assertEqual(c in u, c in self.d or c in self.otherword) self.assertEqual(self.s, self.thetype(self.word)) #self.assertEqual(type(u), self.thetype) self.assertIsInstance(u, self.thetype) self.assertRaises(PassThru, self.s.union, check_pass_thru()) self.assertRaises(TypeError, self.s.union, [[]]) for C in set, frozenset, dict.fromkeys, str, unicode, list, tuple: self.assertEqual(self.thetype('abcba').union(C('cdc')), set('abcd')) self.assertEqual(self.thetype('abcba').union(C('efgfe')), set('abcefg')) self.assertEqual(self.thetype('abcba').union(C('ccb')), set('abc')) self.assertEqual(self.thetype('abcba').union(C('ef')), set('abcef')) self.assertEqual(self.thetype('abcba').union(C('ef'), C('fg')), set('abcefg')) # Issue #6573 x = self.thetype() self.assertEqual(x.union(set([1]), x, set([2])), self.thetype([1, 2])) def test_symmetric_difference(self): i = self.s.symmetric_difference(self.otherword) for c in self.letters: self.assertEqual(c in i, (c in self.d) ^ (c in self.otherword)) self.assertEqual(self.s, self.thetype(self.word)) #self.assertEqual(type(i), self.thetype) self.assertIsInstance(i, self.thetype) self.assertRaises(PassThru, self.s.symmetric_difference, check_pass_thru()) self.assertRaises(TypeError, self.s.symmetric_difference, [[]]) for C in set, frozenset, dict.fromkeys, str, unicode, list, tuple: self.assertEqual(self.thetype('abcba').symmetric_difference(C('cdc')), set('abd')) self.assertEqual(self.thetype('abcba').symmetric_difference(C('efgfe')), set('abcefg')) self.assertEqual(self.thetype('abcba').symmetric_difference(C('ccb')), set('a')) self.assertEqual(self.thetype('abcba').symmetric_difference(C('ef')), set('abcef')) def test_difference(self): i = self.s.difference(self.otherword) for c in self.letters: self.assertEqual(c in i, c in self.d and c not in self.otherword) self.assertEqual(self.s, self.thetype(self.word)) #self.assertEqual(type(i), self.thetype) self.assertIsInstance(i, self.thetype) self.assertRaises(PassThru, self.s.difference, check_pass_thru()) self.assertRaises(TypeError, self.s.difference, [[]]) for C in set, frozenset, dict.fromkeys, str, unicode, list, tuple: self.assertEqual(self.thetype('abcba').difference(C('cdc')), set('ab')) self.assertEqual(self.thetype('abcba').difference(C('efgfe')), set('abc')) self.assertEqual(self.thetype('abcba').difference(C('ccb')), set('a')) self.assertEqual(self.thetype('abcba').difference(C('ef')), set('abc')) self.assertEqual(self.thetype('abcba').difference(), set('abc')) self.assertEqual(self.thetype('abcba').difference(C('a'), C('b')), set('c')) def test_intersection(self): i = self.s.intersection(self.otherword) for c in self.letters: self.assertEqual(c in i, c in self.d and c in self.otherword) self.assertEqual(self.s, self.thetype(self.word)) #self.assertEqual(type(i), self.thetype) self.assertIsInstance(i, self.thetype) self.assertRaises(PassThru, self.s.intersection, check_pass_thru()) for C in set, frozenset, dict.fromkeys, str, unicode, list, tuple: self.assertEqual(self.thetype('abcba').intersection(C('cdc')), set('cc')) self.assertEqual(self.thetype('abcba').intersection(C('efgfe')), set('')) self.assertEqual(self.thetype('abcba').intersection(C('ccb')), set('bc')) self.assertEqual(self.thetype('abcba').intersection(C('ef')), set('')) self.assertEqual(self.thetype('abcba').intersection(C('cbcf'), C('bag')), set('b')) s = self.thetype('abcba') z = s.intersection() if self.thetype == frozenset(): self.assertEqual(id(s), id(z)) else: self.assertNotEqual(id(s), id(z)) def test_copy(self): dup = self.s.copy() self.assertEqual(id(self.s), id(dup)) class TestMutableSet (TestSet, test.test_set.TestSet): thetype = NSMutableSet basetype = NSMutableSet def test_copy(self): dup = self.s.copy() self.assertEqual(self.s, dup) self.assertNotEqual(id(self.s), id(dup)) #self.assertEqual(type(dup), self.basetype) self.assertIsInstance(dup, self.basetype) # Tests from 'TestSet' def test_init(self): pass def test_hash(self): pass def test_weakref(self): pass class TestBasicOpsEmpty (test.test_set.TestBasicOps): def setUp(self): self.case = "empty set" self.values = [] self.set = NSMutableSet(self.values) self.dup = NSMutableSet(self.values) self.length = 0 self.repr = "{(\n)}" def test_pickling(self): pass class TestBasicOpsSingleton (test.test_set.TestBasicOps): def setUp(self): self.case = "unit set (number)" self.values = [3] self.set = NSMutableSet(self.values) self.dup = NSMutableSet(self.values) self.length = 1 self.repr = "{(\n 3\n)}" test.test_set.set = NSMutableSet def tearDown(self): del test.test_set.set def test_pickling(self): pass class TestBasicOpsTuple (test.test_set.TestBasicOps): def setUp(self): self.case = "unit set (tuple)" self.values = [(0, "zero")] self.set = NSMutableSet(self.values) self.dup = NSMutableSet(self.values) self.length = 1 self.repr = "{(\n (\n 0,\n zero\n )\n)}" test.test_set.set = NSMutableSet def tearDown(self): del test.test_set.set def test_pickling(self): pass class TestBasicOpsTriple (test.test_set.TestBasicOps): def setUp(self): self.case = "triple set" self.values = [0, "zero", operator.add] self.set = NSMutableSet(self.values) self.dup = NSMutableSet(self.values) self.length = 3 self.repr = None test.test_set.set = NSMutableSet def tearDown(self): del test.test_set.set def test_pickling(self): pass class TestBinaryOps (test.test_set.TestBinaryOps): def setUp(self): self.set = NSMutableSet((2, 4, 6)) class TestUpdateOps (test.test_set.TestUpdateOps): def setUp(self): self.set = NSMutableSet((2, 4, 6)) class TestMutate (test.test_set.TestMutate): def setUp(self): self.values = ["a", "b", "c"] self.set = NSMutableSet(self.values) test.test_set.set = NSMutableSet def tearDown(self): del test.test_set.set class TestSubsetEqualEmpty (test.test_set.TestSubsetEqualEmpty): left = NSMutableSet() right = NSMutableSet() class TestSubsetEqualNonEmpty (test.test_set.TestSubsetEqualNonEmpty): left = NSMutableSet([1, 2]) right = NSMutableSet([1, 2]) class TestSubsetPartial (test.test_set.TestSubsetPartial): left = NSMutableSet([1]) right = NSMutableSet([1, 2]) class TestOnlySetsNumeric (test.test_set.TestOnlySetsNumeric): def setUp(self): self.set = NSMutableSet((1, 2, 3)) self.other = 19 self.otherIsIterable = False class TestOnlySetsOperator (test.test_set.TestOnlySetsOperator): def setUp(self): self.set = NSMutableSet((1, 2, 3)) self.other = operator.add self.otherIsIterable = False class TestOnlySetsTuple (test.test_set.TestOnlySetsTuple): def setUp(self): self.set = NSMutableSet((1, 2, 3)) self.other = (2, 4, 6) self.otherIsIterable = True class TestOnlySetsString (test.test_set.TestOnlySetsString): def setUp(self): def gen(): for i in xrange(0, 10, 2): yield i self.set = NSMutableSet((1, 2, 3)) self.other = gen() self.otherIsIterable = True class TestIdentities (test.test_set.TestIdentities): def setUp(self): self.a = NSMutableSet('abracadabra') self.b = NSMutableSet('alacazam') class TestVariousIteratorArgs (test.test_set.TestVariousIteratorArgs): def setUp(self): test.test_set.set = NSMutableSet def tearDown(self): del test.test_set.set def test_inplace_methods(self): for data in ("123", "", range(1000), ('do', 1.2), xrange(2000,2200,5), 'december'): for methname in ('update', 'intersection_update', 'difference_update', 'symmetric_difference_update'): for g in (test.test_set.G, test.test_set.I, test.test_set.Ig, test.test_set.S, test.test_set.L, test.test_set.R): #s = set('january') s = NSMutableSet('january') #t = s.copy() t = s.mutableCopy() getattr(s, methname)(list(g(data))) getattr(t, methname)(g(data)) self.assertEqual(sorted(s, key=repr), sorted(t, key=repr)) self.assertRaises(TypeError, getattr(set('january'), methname), test.test_set.X(data)) self.assertRaises(TypeError, getattr(set('january'), methname), test.test_set.N(data)) self.assertRaises(ZeroDivisionError, getattr(set('january'), methname), test.test_set.E(data)) class TestGraphs (test.test_set.TestGraphs): def setUp(self): test.test_set.set = NSMutableSet def tearDown(self): del test.test_set.set if __name__ == "__main__": main()