test_sets.py 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861
  1. import unittest, operator, copy, pickle, random
  2. from test import test_support
  3. test_support.import_module("sets", deprecated=True)
  4. from sets import Set, ImmutableSet
  5. empty_set = Set()
  6. #==============================================================================
  7. class TestBasicOps(unittest.TestCase):
  8. def test_repr(self):
  9. if self.repr is not None:
  10. self.assertEqual(repr(self.set), self.repr)
  11. def test_length(self):
  12. self.assertEqual(len(self.set), self.length)
  13. def test_self_equality(self):
  14. self.assertEqual(self.set, self.set)
  15. def test_equivalent_equality(self):
  16. self.assertEqual(self.set, self.dup)
  17. def test_copy(self):
  18. self.assertEqual(self.set.copy(), self.dup)
  19. def test_self_union(self):
  20. result = self.set | self.set
  21. self.assertEqual(result, self.dup)
  22. def test_empty_union(self):
  23. result = self.set | empty_set
  24. self.assertEqual(result, self.dup)
  25. def test_union_empty(self):
  26. result = empty_set | self.set
  27. self.assertEqual(result, self.dup)
  28. def test_self_intersection(self):
  29. result = self.set & self.set
  30. self.assertEqual(result, self.dup)
  31. def test_empty_intersection(self):
  32. result = self.set & empty_set
  33. self.assertEqual(result, empty_set)
  34. def test_intersection_empty(self):
  35. result = empty_set & self.set
  36. self.assertEqual(result, empty_set)
  37. def test_self_symmetric_difference(self):
  38. result = self.set ^ self.set
  39. self.assertEqual(result, empty_set)
  40. def checkempty_symmetric_difference(self):
  41. result = self.set ^ empty_set
  42. self.assertEqual(result, self.set)
  43. def test_self_difference(self):
  44. result = self.set - self.set
  45. self.assertEqual(result, empty_set)
  46. def test_empty_difference(self):
  47. result = self.set - empty_set
  48. self.assertEqual(result, self.dup)
  49. def test_empty_difference_rev(self):
  50. result = empty_set - self.set
  51. self.assertEqual(result, empty_set)
  52. def test_iteration(self):
  53. for v in self.set:
  54. self.assertIn(v, self.values)
  55. def test_pickling(self):
  56. for proto in range(pickle.HIGHEST_PROTOCOL + 1):
  57. p = pickle.dumps(self.set, proto)
  58. copy = pickle.loads(p)
  59. self.assertEqual(self.set, copy,
  60. "%s != %s" % (self.set, copy))
  61. #------------------------------------------------------------------------------
  62. class TestBasicOpsEmpty(TestBasicOps):
  63. def setUp(self):
  64. self.case = "empty set"
  65. self.values = []
  66. self.set = Set(self.values)
  67. self.dup = Set(self.values)
  68. self.length = 0
  69. self.repr = "Set([])"
  70. #------------------------------------------------------------------------------
  71. class TestBasicOpsSingleton(TestBasicOps):
  72. def setUp(self):
  73. self.case = "unit set (number)"
  74. self.values = [3]
  75. self.set = Set(self.values)
  76. self.dup = Set(self.values)
  77. self.length = 1
  78. self.repr = "Set([3])"
  79. def test_in(self):
  80. self.assertTrue(3 in self.set)
  81. def test_not_in(self):
  82. self.assertTrue(2 not in self.set)
  83. #------------------------------------------------------------------------------
  84. class TestBasicOpsTuple(TestBasicOps):
  85. def setUp(self):
  86. self.case = "unit set (tuple)"
  87. self.values = [(0, "zero")]
  88. self.set = Set(self.values)
  89. self.dup = Set(self.values)
  90. self.length = 1
  91. self.repr = "Set([(0, 'zero')])"
  92. def test_in(self):
  93. self.assertTrue((0, "zero") in self.set)
  94. def test_not_in(self):
  95. self.assertTrue(9 not in self.set)
  96. #------------------------------------------------------------------------------
  97. class TestBasicOpsTriple(TestBasicOps):
  98. def setUp(self):
  99. self.case = "triple set"
  100. self.values = [0, "zero", operator.add]
  101. self.set = Set(self.values)
  102. self.dup = Set(self.values)
  103. self.length = 3
  104. self.repr = None
  105. #==============================================================================
  106. def baditer():
  107. raise TypeError
  108. yield True
  109. def gooditer():
  110. yield True
  111. class TestExceptionPropagation(unittest.TestCase):
  112. """SF 628246: Set constructor should not trap iterator TypeErrors"""
  113. def test_instanceWithException(self):
  114. self.assertRaises(TypeError, Set, baditer())
  115. def test_instancesWithoutException(self):
  116. # All of these iterables should load without exception.
  117. Set([1,2,3])
  118. Set((1,2,3))
  119. Set({'one':1, 'two':2, 'three':3})
  120. Set(xrange(3))
  121. Set('abc')
  122. Set(gooditer())
  123. #==============================================================================
  124. class TestSetOfSets(unittest.TestCase):
  125. def test_constructor(self):
  126. inner = Set([1])
  127. outer = Set([inner])
  128. element = outer.pop()
  129. self.assertEqual(type(element), ImmutableSet)
  130. outer.add(inner) # Rebuild set of sets with .add method
  131. outer.remove(inner)
  132. self.assertEqual(outer, Set()) # Verify that remove worked
  133. outer.discard(inner) # Absence of KeyError indicates working fine
  134. #==============================================================================
  135. class TestBinaryOps(unittest.TestCase):
  136. def setUp(self):
  137. self.set = Set((2, 4, 6))
  138. def test_eq(self): # SF bug 643115
  139. self.assertEqual(self.set, Set({2:1,4:3,6:5}))
  140. def test_union_subset(self):
  141. result = self.set | Set([2])
  142. self.assertEqual(result, Set((2, 4, 6)))
  143. def test_union_superset(self):
  144. result = self.set | Set([2, 4, 6, 8])
  145. self.assertEqual(result, Set([2, 4, 6, 8]))
  146. def test_union_overlap(self):
  147. result = self.set | Set([3, 4, 5])
  148. self.assertEqual(result, Set([2, 3, 4, 5, 6]))
  149. def test_union_non_overlap(self):
  150. result = self.set | Set([8])
  151. self.assertEqual(result, Set([2, 4, 6, 8]))
  152. def test_intersection_subset(self):
  153. result = self.set & Set((2, 4))
  154. self.assertEqual(result, Set((2, 4)))
  155. def test_intersection_superset(self):
  156. result = self.set & Set([2, 4, 6, 8])
  157. self.assertEqual(result, Set([2, 4, 6]))
  158. def test_intersection_overlap(self):
  159. result = self.set & Set([3, 4, 5])
  160. self.assertEqual(result, Set([4]))
  161. def test_intersection_non_overlap(self):
  162. result = self.set & Set([8])
  163. self.assertEqual(result, empty_set)
  164. def test_sym_difference_subset(self):
  165. result = self.set ^ Set((2, 4))
  166. self.assertEqual(result, Set([6]))
  167. def test_sym_difference_superset(self):
  168. result = self.set ^ Set((2, 4, 6, 8))
  169. self.assertEqual(result, Set([8]))
  170. def test_sym_difference_overlap(self):
  171. result = self.set ^ Set((3, 4, 5))
  172. self.assertEqual(result, Set([2, 3, 5, 6]))
  173. def test_sym_difference_non_overlap(self):
  174. result = self.set ^ Set([8])
  175. self.assertEqual(result, Set([2, 4, 6, 8]))
  176. def test_cmp(self):
  177. a, b = Set('a'), Set('b')
  178. self.assertRaises(TypeError, cmp, a, b)
  179. # You can view this as a buglet: cmp(a, a) does not raise TypeError,
  180. # because __eq__ is tried before __cmp__, and a.__eq__(a) returns True,
  181. # which Python thinks is good enough to synthesize a cmp() result
  182. # without calling __cmp__.
  183. self.assertEqual(cmp(a, a), 0)
  184. self.assertRaises(TypeError, cmp, a, 12)
  185. self.assertRaises(TypeError, cmp, "abc", a)
  186. def test_inplace_on_self(self):
  187. t = self.set.copy()
  188. t |= t
  189. self.assertEqual(t, self.set)
  190. t &= t
  191. self.assertEqual(t, self.set)
  192. t -= t
  193. self.assertEqual(len(t), 0)
  194. t = self.set.copy()
  195. t ^= t
  196. self.assertEqual(len(t), 0)
  197. #==============================================================================
  198. class TestUpdateOps(unittest.TestCase):
  199. def setUp(self):
  200. self.set = Set((2, 4, 6))
  201. def test_union_subset(self):
  202. self.set |= Set([2])
  203. self.assertEqual(self.set, Set((2, 4, 6)))
  204. def test_union_superset(self):
  205. self.set |= Set([2, 4, 6, 8])
  206. self.assertEqual(self.set, Set([2, 4, 6, 8]))
  207. def test_union_overlap(self):
  208. self.set |= Set([3, 4, 5])
  209. self.assertEqual(self.set, Set([2, 3, 4, 5, 6]))
  210. def test_union_non_overlap(self):
  211. self.set |= Set([8])
  212. self.assertEqual(self.set, Set([2, 4, 6, 8]))
  213. def test_union_method_call(self):
  214. self.set.union_update(Set([3, 4, 5]))
  215. self.assertEqual(self.set, Set([2, 3, 4, 5, 6]))
  216. def test_intersection_subset(self):
  217. self.set &= Set((2, 4))
  218. self.assertEqual(self.set, Set((2, 4)))
  219. def test_intersection_superset(self):
  220. self.set &= Set([2, 4, 6, 8])
  221. self.assertEqual(self.set, Set([2, 4, 6]))
  222. def test_intersection_overlap(self):
  223. self.set &= Set([3, 4, 5])
  224. self.assertEqual(self.set, Set([4]))
  225. def test_intersection_non_overlap(self):
  226. self.set &= Set([8])
  227. self.assertEqual(self.set, empty_set)
  228. def test_intersection_method_call(self):
  229. self.set.intersection_update(Set([3, 4, 5]))
  230. self.assertEqual(self.set, Set([4]))
  231. def test_sym_difference_subset(self):
  232. self.set ^= Set((2, 4))
  233. self.assertEqual(self.set, Set([6]))
  234. def test_sym_difference_superset(self):
  235. self.set ^= Set((2, 4, 6, 8))
  236. self.assertEqual(self.set, Set([8]))
  237. def test_sym_difference_overlap(self):
  238. self.set ^= Set((3, 4, 5))
  239. self.assertEqual(self.set, Set([2, 3, 5, 6]))
  240. def test_sym_difference_non_overlap(self):
  241. self.set ^= Set([8])
  242. self.assertEqual(self.set, Set([2, 4, 6, 8]))
  243. def test_sym_difference_method_call(self):
  244. self.set.symmetric_difference_update(Set([3, 4, 5]))
  245. self.assertEqual(self.set, Set([2, 3, 5, 6]))
  246. def test_difference_subset(self):
  247. self.set -= Set((2, 4))
  248. self.assertEqual(self.set, Set([6]))
  249. def test_difference_superset(self):
  250. self.set -= Set((2, 4, 6, 8))
  251. self.assertEqual(self.set, Set([]))
  252. def test_difference_overlap(self):
  253. self.set -= Set((3, 4, 5))
  254. self.assertEqual(self.set, Set([2, 6]))
  255. def test_difference_non_overlap(self):
  256. self.set -= Set([8])
  257. self.assertEqual(self.set, Set([2, 4, 6]))
  258. def test_difference_method_call(self):
  259. self.set.difference_update(Set([3, 4, 5]))
  260. self.assertEqual(self.set, Set([2, 6]))
  261. #==============================================================================
  262. class TestMutate(unittest.TestCase):
  263. def setUp(self):
  264. self.values = ["a", "b", "c"]
  265. self.set = Set(self.values)
  266. def test_add_present(self):
  267. self.set.add("c")
  268. self.assertEqual(self.set, Set("abc"))
  269. def test_add_absent(self):
  270. self.set.add("d")
  271. self.assertEqual(self.set, Set("abcd"))
  272. def test_add_until_full(self):
  273. tmp = Set()
  274. expected_len = 0
  275. for v in self.values:
  276. tmp.add(v)
  277. expected_len += 1
  278. self.assertEqual(len(tmp), expected_len)
  279. self.assertEqual(tmp, self.set)
  280. def test_remove_present(self):
  281. self.set.remove("b")
  282. self.assertEqual(self.set, Set("ac"))
  283. def test_remove_absent(self):
  284. try:
  285. self.set.remove("d")
  286. self.fail("Removing missing element should have raised LookupError")
  287. except LookupError:
  288. pass
  289. def test_remove_until_empty(self):
  290. expected_len = len(self.set)
  291. for v in self.values:
  292. self.set.remove(v)
  293. expected_len -= 1
  294. self.assertEqual(len(self.set), expected_len)
  295. def test_discard_present(self):
  296. self.set.discard("c")
  297. self.assertEqual(self.set, Set("ab"))
  298. def test_discard_absent(self):
  299. self.set.discard("d")
  300. self.assertEqual(self.set, Set("abc"))
  301. def test_clear(self):
  302. self.set.clear()
  303. self.assertEqual(len(self.set), 0)
  304. def test_pop(self):
  305. popped = {}
  306. while self.set:
  307. popped[self.set.pop()] = None
  308. self.assertEqual(len(popped), len(self.values))
  309. for v in self.values:
  310. self.assertIn(v, popped)
  311. def test_update_empty_tuple(self):
  312. self.set.union_update(())
  313. self.assertEqual(self.set, Set(self.values))
  314. def test_update_unit_tuple_overlap(self):
  315. self.set.union_update(("a",))
  316. self.assertEqual(self.set, Set(self.values))
  317. def test_update_unit_tuple_non_overlap(self):
  318. self.set.union_update(("a", "z"))
  319. self.assertEqual(self.set, Set(self.values + ["z"]))
  320. #==============================================================================
  321. class TestSubsets(unittest.TestCase):
  322. case2method = {"<=": "issubset",
  323. ">=": "issuperset",
  324. }
  325. reverse = {"==": "==",
  326. "!=": "!=",
  327. "<": ">",
  328. ">": "<",
  329. "<=": ">=",
  330. ">=": "<=",
  331. }
  332. def test_issubset(self):
  333. x = self.left
  334. y = self.right
  335. for case in "!=", "==", "<", "<=", ">", ">=":
  336. expected = case in self.cases
  337. # Test the binary infix spelling.
  338. result = eval("x" + case + "y", locals())
  339. self.assertEqual(result, expected)
  340. # Test the "friendly" method-name spelling, if one exists.
  341. if case in TestSubsets.case2method:
  342. method = getattr(x, TestSubsets.case2method[case])
  343. result = method(y)
  344. self.assertEqual(result, expected)
  345. # Now do the same for the operands reversed.
  346. rcase = TestSubsets.reverse[case]
  347. result = eval("y" + rcase + "x", locals())
  348. self.assertEqual(result, expected)
  349. if rcase in TestSubsets.case2method:
  350. method = getattr(y, TestSubsets.case2method[rcase])
  351. result = method(x)
  352. self.assertEqual(result, expected)
  353. #------------------------------------------------------------------------------
  354. class TestSubsetEqualEmpty(TestSubsets):
  355. left = Set()
  356. right = Set()
  357. name = "both empty"
  358. cases = "==", "<=", ">="
  359. #------------------------------------------------------------------------------
  360. class TestSubsetEqualNonEmpty(TestSubsets):
  361. left = Set([1, 2])
  362. right = Set([1, 2])
  363. name = "equal pair"
  364. cases = "==", "<=", ">="
  365. #------------------------------------------------------------------------------
  366. class TestSubsetEmptyNonEmpty(TestSubsets):
  367. left = Set()
  368. right = Set([1, 2])
  369. name = "one empty, one non-empty"
  370. cases = "!=", "<", "<="
  371. #------------------------------------------------------------------------------
  372. class TestSubsetPartial(TestSubsets):
  373. left = Set([1])
  374. right = Set([1, 2])
  375. name = "one a non-empty proper subset of other"
  376. cases = "!=", "<", "<="
  377. #------------------------------------------------------------------------------
  378. class TestSubsetNonOverlap(TestSubsets):
  379. left = Set([1])
  380. right = Set([2])
  381. name = "neither empty, neither contains"
  382. cases = "!="
  383. #==============================================================================
  384. class TestOnlySetsInBinaryOps(unittest.TestCase):
  385. def test_eq_ne(self):
  386. # Unlike the others, this is testing that == and != *are* allowed.
  387. self.assertEqual(self.other == self.set, False)
  388. self.assertEqual(self.set == self.other, False)
  389. self.assertEqual(self.other != self.set, True)
  390. self.assertEqual(self.set != self.other, True)
  391. def test_ge_gt_le_lt(self):
  392. self.assertRaises(TypeError, lambda: self.set < self.other)
  393. self.assertRaises(TypeError, lambda: self.set <= self.other)
  394. self.assertRaises(TypeError, lambda: self.set > self.other)
  395. self.assertRaises(TypeError, lambda: self.set >= self.other)
  396. self.assertRaises(TypeError, lambda: self.other < self.set)
  397. self.assertRaises(TypeError, lambda: self.other <= self.set)
  398. self.assertRaises(TypeError, lambda: self.other > self.set)
  399. self.assertRaises(TypeError, lambda: self.other >= self.set)
  400. def test_union_update_operator(self):
  401. try:
  402. self.set |= self.other
  403. except TypeError:
  404. pass
  405. else:
  406. self.fail("expected TypeError")
  407. def test_union_update(self):
  408. if self.otherIsIterable:
  409. self.set.union_update(self.other)
  410. else:
  411. self.assertRaises(TypeError, self.set.union_update, self.other)
  412. def test_union(self):
  413. self.assertRaises(TypeError, lambda: self.set | self.other)
  414. self.assertRaises(TypeError, lambda: self.other | self.set)
  415. if self.otherIsIterable:
  416. self.set.union(self.other)
  417. else:
  418. self.assertRaises(TypeError, self.set.union, self.other)
  419. def test_intersection_update_operator(self):
  420. try:
  421. self.set &= self.other
  422. except TypeError:
  423. pass
  424. else:
  425. self.fail("expected TypeError")
  426. def test_intersection_update(self):
  427. if self.otherIsIterable:
  428. self.set.intersection_update(self.other)
  429. else:
  430. self.assertRaises(TypeError,
  431. self.set.intersection_update,
  432. self.other)
  433. def test_intersection(self):
  434. self.assertRaises(TypeError, lambda: self.set & self.other)
  435. self.assertRaises(TypeError, lambda: self.other & self.set)
  436. if self.otherIsIterable:
  437. self.set.intersection(self.other)
  438. else:
  439. self.assertRaises(TypeError, self.set.intersection, self.other)
  440. def test_sym_difference_update_operator(self):
  441. try:
  442. self.set ^= self.other
  443. except TypeError:
  444. pass
  445. else:
  446. self.fail("expected TypeError")
  447. def test_sym_difference_update(self):
  448. if self.otherIsIterable:
  449. self.set.symmetric_difference_update(self.other)
  450. else:
  451. self.assertRaises(TypeError,
  452. self.set.symmetric_difference_update,
  453. self.other)
  454. def test_sym_difference(self):
  455. self.assertRaises(TypeError, lambda: self.set ^ self.other)
  456. self.assertRaises(TypeError, lambda: self.other ^ self.set)
  457. if self.otherIsIterable:
  458. self.set.symmetric_difference(self.other)
  459. else:
  460. self.assertRaises(TypeError, self.set.symmetric_difference, self.other)
  461. def test_difference_update_operator(self):
  462. try:
  463. self.set -= self.other
  464. except TypeError:
  465. pass
  466. else:
  467. self.fail("expected TypeError")
  468. def test_difference_update(self):
  469. if self.otherIsIterable:
  470. self.set.difference_update(self.other)
  471. else:
  472. self.assertRaises(TypeError,
  473. self.set.difference_update,
  474. self.other)
  475. def test_difference(self):
  476. self.assertRaises(TypeError, lambda: self.set - self.other)
  477. self.assertRaises(TypeError, lambda: self.other - self.set)
  478. if self.otherIsIterable:
  479. self.set.difference(self.other)
  480. else:
  481. self.assertRaises(TypeError, self.set.difference, self.other)
  482. #------------------------------------------------------------------------------
  483. class TestOnlySetsNumeric(TestOnlySetsInBinaryOps):
  484. def setUp(self):
  485. self.set = Set((1, 2, 3))
  486. self.other = 19
  487. self.otherIsIterable = False
  488. #------------------------------------------------------------------------------
  489. class TestOnlySetsDict(TestOnlySetsInBinaryOps):
  490. def setUp(self):
  491. self.set = Set((1, 2, 3))
  492. self.other = {1:2, 3:4}
  493. self.otherIsIterable = True
  494. #------------------------------------------------------------------------------
  495. class TestOnlySetsOperator(TestOnlySetsInBinaryOps):
  496. def setUp(self):
  497. self.set = Set((1, 2, 3))
  498. self.other = operator.add
  499. self.otherIsIterable = False
  500. def test_ge_gt_le_lt(self):
  501. with test_support.check_py3k_warnings():
  502. super(TestOnlySetsOperator, self).test_ge_gt_le_lt()
  503. #------------------------------------------------------------------------------
  504. class TestOnlySetsTuple(TestOnlySetsInBinaryOps):
  505. def setUp(self):
  506. self.set = Set((1, 2, 3))
  507. self.other = (2, 4, 6)
  508. self.otherIsIterable = True
  509. #------------------------------------------------------------------------------
  510. class TestOnlySetsString(TestOnlySetsInBinaryOps):
  511. def setUp(self):
  512. self.set = Set((1, 2, 3))
  513. self.other = 'abc'
  514. self.otherIsIterable = True
  515. #------------------------------------------------------------------------------
  516. class TestOnlySetsGenerator(TestOnlySetsInBinaryOps):
  517. def setUp(self):
  518. def gen():
  519. for i in xrange(0, 10, 2):
  520. yield i
  521. self.set = Set((1, 2, 3))
  522. self.other = gen()
  523. self.otherIsIterable = True
  524. #------------------------------------------------------------------------------
  525. class TestOnlySetsofSets(TestOnlySetsInBinaryOps):
  526. def setUp(self):
  527. self.set = Set((1, 2, 3))
  528. self.other = [Set('ab'), ImmutableSet('cd')]
  529. self.otherIsIterable = True
  530. #==============================================================================
  531. class TestCopying(unittest.TestCase):
  532. def test_copy(self):
  533. dup = self.set.copy()
  534. self.assertEqual(len(dup), len(self.set))
  535. dup_list = sorted(dup)
  536. set_list = sorted(self.set)
  537. self.assertEqual(len(dup_list), len(set_list))
  538. for i, el in enumerate(dup_list):
  539. self.assertIs(el, set_list[i])
  540. def test_deep_copy(self):
  541. dup = copy.deepcopy(self.set)
  542. self.assertSetEqual(dup, self.set)
  543. #------------------------------------------------------------------------------
  544. class TestCopyingEmpty(TestCopying):
  545. def setUp(self):
  546. self.set = Set()
  547. #------------------------------------------------------------------------------
  548. class TestCopyingSingleton(TestCopying):
  549. def setUp(self):
  550. self.set = Set(["hello"])
  551. #------------------------------------------------------------------------------
  552. class TestCopyingTriple(TestCopying):
  553. def setUp(self):
  554. self.set = Set(["zero", 0, None])
  555. def test_copy(self):
  556. with test_support.check_py3k_warnings():
  557. super(TestCopyingTriple, self).test_copy()
  558. #------------------------------------------------------------------------------
  559. class TestCopyingTuple(TestCopying):
  560. def setUp(self):
  561. self.set = Set([(1, 2)])
  562. #------------------------------------------------------------------------------
  563. class TestCopyingNested(TestCopying):
  564. def setUp(self):
  565. self.set = Set([((1, 2), (3, 4))])
  566. #==============================================================================
  567. class TestIdentities(unittest.TestCase):
  568. def setUp(self):
  569. self.a = Set([random.randrange(100) for i in xrange(50)])
  570. self.b = Set([random.randrange(100) for i in xrange(50)])
  571. def test_binopsVsSubsets(self):
  572. a, b = self.a, self.b
  573. self.assertTrue(a - b <= a)
  574. self.assertTrue(b - a <= b)
  575. self.assertTrue(a & b <= a)
  576. self.assertTrue(a & b <= b)
  577. self.assertTrue(a | b >= a)
  578. self.assertTrue(a | b >= b)
  579. self.assertTrue(a ^ b <= a | b)
  580. def test_commutativity(self):
  581. a, b = self.a, self.b
  582. self.assertEqual(a&b, b&a)
  583. self.assertEqual(a|b, b|a)
  584. self.assertEqual(a^b, b^a)
  585. if a != b:
  586. self.assertNotEqual(a-b, b-a)
  587. def test_reflexsive_relations(self):
  588. a, zero = self.a, Set()
  589. self.assertEqual(a ^ a, zero)
  590. self.assertEqual(a - a, zero)
  591. self.assertEqual(a | a, a)
  592. self.assertEqual(a & a, a)
  593. self.assertTrue(a <= a)
  594. self.assertTrue(a >= a)
  595. self.assertTrue(a == a)
  596. def test_summations(self):
  597. # check that sums of parts equal the whole
  598. a, b = self.a, self.b
  599. self.assertEqual((a-b)|(a&b)|(b-a), a|b)
  600. self.assertEqual((a&b)|(a^b), a|b)
  601. self.assertEqual(a|(b-a), a|b)
  602. self.assertEqual((a-b)|b, a|b)
  603. self.assertEqual((a-b)|(a&b), a)
  604. self.assertEqual((b-a)|(a&b), b)
  605. self.assertEqual((a-b)|(b-a), a^b)
  606. def test_exclusion(self):
  607. # check that inverse operations do not overlap
  608. a, b, zero = self.a, self.b, Set()
  609. self.assertEqual((a-b)&b, zero)
  610. self.assertEqual((b-a)&a, zero)
  611. self.assertEqual((a&b)&(a^b), zero)
  612. def test_cardinality_relations(self):
  613. a, b = self.a, self.b
  614. self.assertEqual(len(a), len(a-b) + len(a&b))
  615. self.assertEqual(len(b), len(b-a) + len(a&b))
  616. self.assertEqual(len(a^b), len(a-b) + len(b-a))
  617. self.assertEqual(len(a|b), len(a-b) + len(a&b) + len(b-a))
  618. self.assertEqual(len(a^b) + len(a&b), len(a|b))
  619. #==============================================================================
  620. libreftest = """
  621. Example from the Library Reference: Doc/lib/libsets.tex
  622. >>> from sets import Set as Base # override _repr to get sorted output
  623. >>> class Set(Base):
  624. ... def _repr(self):
  625. ... return Base._repr(self, sorted=True)
  626. >>> engineers = Set(['John', 'Jane', 'Jack', 'Janice'])
  627. >>> programmers = Set(['Jack', 'Sam', 'Susan', 'Janice'])
  628. >>> managers = Set(['Jane', 'Jack', 'Susan', 'Zack'])
  629. >>> employees = engineers | programmers | managers # union
  630. >>> engineering_management = engineers & managers # intersection
  631. >>> fulltime_management = managers - engineers - programmers # difference
  632. >>> engineers.add('Marvin')
  633. >>> print engineers
  634. Set(['Jack', 'Jane', 'Janice', 'John', 'Marvin'])
  635. >>> employees.issuperset(engineers) # superset test
  636. False
  637. >>> employees.union_update(engineers) # update from another set
  638. >>> employees.issuperset(engineers)
  639. True
  640. >>> for group in [engineers, programmers, managers, employees]:
  641. ... group.discard('Susan') # unconditionally remove element
  642. ... print group
  643. ...
  644. Set(['Jack', 'Jane', 'Janice', 'John', 'Marvin'])
  645. Set(['Jack', 'Janice', 'Sam'])
  646. Set(['Jack', 'Jane', 'Zack'])
  647. Set(['Jack', 'Jane', 'Janice', 'John', 'Marvin', 'Sam', 'Zack'])
  648. """
  649. #==============================================================================
  650. __test__ = {'libreftest' : libreftest}
  651. def test_main(verbose=None):
  652. import doctest
  653. from test import test_sets
  654. test_support.run_unittest(
  655. TestSetOfSets,
  656. TestExceptionPropagation,
  657. TestBasicOpsEmpty,
  658. TestBasicOpsSingleton,
  659. TestBasicOpsTuple,
  660. TestBasicOpsTriple,
  661. TestBinaryOps,
  662. TestUpdateOps,
  663. TestMutate,
  664. TestSubsetEqualEmpty,
  665. TestSubsetEqualNonEmpty,
  666. TestSubsetEmptyNonEmpty,
  667. TestSubsetPartial,
  668. TestSubsetNonOverlap,
  669. TestOnlySetsNumeric,
  670. TestOnlySetsDict,
  671. TestOnlySetsOperator,
  672. TestOnlySetsTuple,
  673. TestOnlySetsString,
  674. TestOnlySetsGenerator,
  675. TestOnlySetsofSets,
  676. TestCopyingEmpty,
  677. TestCopyingSingleton,
  678. TestCopyingTriple,
  679. TestCopyingTuple,
  680. TestCopyingNested,
  681. TestIdentities,
  682. doctest.DocTestSuite(test_sets),
  683. )
  684. if __name__ == "__main__":
  685. test_main(verbose=True)