test_assertions.py 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290
  1. import datetime
  2. import unittest
  3. class Test_Assertions(unittest.TestCase):
  4. def test_AlmostEqual(self):
  5. self.assertAlmostEqual(1.00000001, 1.0)
  6. self.assertNotAlmostEqual(1.0000001, 1.0)
  7. self.assertRaises(self.failureException,
  8. self.assertAlmostEqual, 1.0000001, 1.0)
  9. self.assertRaises(self.failureException,
  10. self.assertNotAlmostEqual, 1.00000001, 1.0)
  11. self.assertAlmostEqual(1.1, 1.0, places=0)
  12. self.assertRaises(self.failureException,
  13. self.assertAlmostEqual, 1.1, 1.0, places=1)
  14. self.assertAlmostEqual(0, .1+.1j, places=0)
  15. self.assertNotAlmostEqual(0, .1+.1j, places=1)
  16. self.assertRaises(self.failureException,
  17. self.assertAlmostEqual, 0, .1+.1j, places=1)
  18. self.assertRaises(self.failureException,
  19. self.assertNotAlmostEqual, 0, .1+.1j, places=0)
  20. self.assertAlmostEqual(float('inf'), float('inf'))
  21. self.assertRaises(self.failureException, self.assertNotAlmostEqual,
  22. float('inf'), float('inf'))
  23. def test_AmostEqualWithDelta(self):
  24. self.assertAlmostEqual(1.1, 1.0, delta=0.5)
  25. self.assertAlmostEqual(1.0, 1.1, delta=0.5)
  26. self.assertNotAlmostEqual(1.1, 1.0, delta=0.05)
  27. self.assertNotAlmostEqual(1.0, 1.1, delta=0.05)
  28. self.assertAlmostEqual(1.0, 1.0, delta=0.5)
  29. self.assertRaises(self.failureException, self.assertNotAlmostEqual,
  30. 1.0, 1.0, delta=0.5)
  31. self.assertRaises(self.failureException, self.assertAlmostEqual,
  32. 1.1, 1.0, delta=0.05)
  33. self.assertRaises(self.failureException, self.assertNotAlmostEqual,
  34. 1.1, 1.0, delta=0.5)
  35. self.assertRaises(TypeError, self.assertAlmostEqual,
  36. 1.1, 1.0, places=2, delta=2)
  37. self.assertRaises(TypeError, self.assertNotAlmostEqual,
  38. 1.1, 1.0, places=2, delta=2)
  39. first = datetime.datetime.now()
  40. second = first + datetime.timedelta(seconds=10)
  41. self.assertAlmostEqual(first, second,
  42. delta=datetime.timedelta(seconds=20))
  43. self.assertNotAlmostEqual(first, second,
  44. delta=datetime.timedelta(seconds=5))
  45. def test_assertRaises(self):
  46. def _raise(e):
  47. raise e
  48. self.assertRaises(KeyError, _raise, KeyError)
  49. self.assertRaises(KeyError, _raise, KeyError("key"))
  50. try:
  51. self.assertRaises(KeyError, lambda: None)
  52. except self.failureException as e:
  53. self.assertIn("KeyError not raised", e.args)
  54. else:
  55. self.fail("assertRaises() didn't fail")
  56. try:
  57. self.assertRaises(KeyError, _raise, ValueError)
  58. except ValueError:
  59. pass
  60. else:
  61. self.fail("assertRaises() didn't let exception pass through")
  62. with self.assertRaises(KeyError) as cm:
  63. try:
  64. raise KeyError
  65. except Exception, e:
  66. raise
  67. self.assertIs(cm.exception, e)
  68. with self.assertRaises(KeyError):
  69. raise KeyError("key")
  70. try:
  71. with self.assertRaises(KeyError):
  72. pass
  73. except self.failureException as e:
  74. self.assertIn("KeyError not raised", e.args)
  75. else:
  76. self.fail("assertRaises() didn't fail")
  77. try:
  78. with self.assertRaises(KeyError):
  79. raise ValueError
  80. except ValueError:
  81. pass
  82. else:
  83. self.fail("assertRaises() didn't let exception pass through")
  84. def testAssertNotRegexpMatches(self):
  85. self.assertNotRegexpMatches('Ala ma kota', r'r+')
  86. try:
  87. self.assertNotRegexpMatches('Ala ma kota', r'k.t', 'Message')
  88. except self.failureException, e:
  89. self.assertIn("'kot'", e.args[0])
  90. self.assertIn('Message', e.args[0])
  91. else:
  92. self.fail('assertNotRegexpMatches should have failed.')
  93. class TestLongMessage(unittest.TestCase):
  94. """Test that the individual asserts honour longMessage.
  95. This actually tests all the message behaviour for
  96. asserts that use longMessage."""
  97. def setUp(self):
  98. class TestableTestFalse(unittest.TestCase):
  99. longMessage = False
  100. failureException = self.failureException
  101. def testTest(self):
  102. pass
  103. class TestableTestTrue(unittest.TestCase):
  104. longMessage = True
  105. failureException = self.failureException
  106. def testTest(self):
  107. pass
  108. self.testableTrue = TestableTestTrue('testTest')
  109. self.testableFalse = TestableTestFalse('testTest')
  110. def testDefault(self):
  111. self.assertFalse(unittest.TestCase.longMessage)
  112. def test_formatMsg(self):
  113. self.assertEqual(self.testableFalse._formatMessage(None, "foo"), "foo")
  114. self.assertEqual(self.testableFalse._formatMessage("foo", "bar"), "foo")
  115. self.assertEqual(self.testableTrue._formatMessage(None, "foo"), "foo")
  116. self.assertEqual(self.testableTrue._formatMessage("foo", "bar"), "bar : foo")
  117. # This blows up if _formatMessage uses string concatenation
  118. self.testableTrue._formatMessage(object(), 'foo')
  119. def test_formatMessage_unicode_error(self):
  120. one = ''.join(chr(i) for i in range(255))
  121. # this used to cause a UnicodeDecodeError constructing msg
  122. self.testableTrue._formatMessage(one, u'\uFFFD')
  123. def assertMessages(self, methodName, args, errors):
  124. def getMethod(i):
  125. useTestableFalse = i < 2
  126. if useTestableFalse:
  127. test = self.testableFalse
  128. else:
  129. test = self.testableTrue
  130. return getattr(test, methodName)
  131. for i, expected_regexp in enumerate(errors):
  132. testMethod = getMethod(i)
  133. kwargs = {}
  134. withMsg = i % 2
  135. if withMsg:
  136. kwargs = {"msg": "oops"}
  137. with self.assertRaisesRegexp(self.failureException,
  138. expected_regexp=expected_regexp):
  139. testMethod(*args, **kwargs)
  140. def testAssertTrue(self):
  141. self.assertMessages('assertTrue', (False,),
  142. ["^False is not true$", "^oops$", "^False is not true$",
  143. "^False is not true : oops$"])
  144. def testAssertFalse(self):
  145. self.assertMessages('assertFalse', (True,),
  146. ["^True is not false$", "^oops$", "^True is not false$",
  147. "^True is not false : oops$"])
  148. def testNotEqual(self):
  149. self.assertMessages('assertNotEqual', (1, 1),
  150. ["^1 == 1$", "^oops$", "^1 == 1$",
  151. "^1 == 1 : oops$"])
  152. def testAlmostEqual(self):
  153. self.assertMessages('assertAlmostEqual', (1, 2),
  154. ["^1 != 2 within 7 places$", "^oops$",
  155. "^1 != 2 within 7 places$", "^1 != 2 within 7 places : oops$"])
  156. def testNotAlmostEqual(self):
  157. self.assertMessages('assertNotAlmostEqual', (1, 1),
  158. ["^1 == 1 within 7 places$", "^oops$",
  159. "^1 == 1 within 7 places$", "^1 == 1 within 7 places : oops$"])
  160. def test_baseAssertEqual(self):
  161. self.assertMessages('_baseAssertEqual', (1, 2),
  162. ["^1 != 2$", "^oops$", "^1 != 2$", "^1 != 2 : oops$"])
  163. def testAssertSequenceEqual(self):
  164. # Error messages are multiline so not testing on full message
  165. # assertTupleEqual and assertListEqual delegate to this method
  166. self.assertMessages('assertSequenceEqual', ([], [None]),
  167. ["\+ \[None\]$", "^oops$", r"\+ \[None\]$",
  168. r"\+ \[None\] : oops$"])
  169. def testAssertSetEqual(self):
  170. self.assertMessages('assertSetEqual', (set(), set([None])),
  171. ["None$", "^oops$", "None$",
  172. "None : oops$"])
  173. def testAssertIn(self):
  174. self.assertMessages('assertIn', (None, []),
  175. ['^None not found in \[\]$', "^oops$",
  176. '^None not found in \[\]$',
  177. '^None not found in \[\] : oops$'])
  178. def testAssertNotIn(self):
  179. self.assertMessages('assertNotIn', (None, [None]),
  180. ['^None unexpectedly found in \[None\]$', "^oops$",
  181. '^None unexpectedly found in \[None\]$',
  182. '^None unexpectedly found in \[None\] : oops$'])
  183. def testAssertDictEqual(self):
  184. self.assertMessages('assertDictEqual', ({}, {'key': 'value'}),
  185. [r"\+ \{'key': 'value'\}$", "^oops$",
  186. "\+ \{'key': 'value'\}$",
  187. "\+ \{'key': 'value'\} : oops$"])
  188. def testAssertDictContainsSubset(self):
  189. self.assertMessages('assertDictContainsSubset', ({'key': 'value'}, {}),
  190. ["^Missing: 'key'$", "^oops$",
  191. "^Missing: 'key'$",
  192. "^Missing: 'key' : oops$"])
  193. def testAssertMultiLineEqual(self):
  194. self.assertMessages('assertMultiLineEqual', ("", "foo"),
  195. [r"\+ foo$", "^oops$",
  196. r"\+ foo$",
  197. r"\+ foo : oops$"])
  198. def testAssertLess(self):
  199. self.assertMessages('assertLess', (2, 1),
  200. ["^2 not less than 1$", "^oops$",
  201. "^2 not less than 1$", "^2 not less than 1 : oops$"])
  202. def testAssertLessEqual(self):
  203. self.assertMessages('assertLessEqual', (2, 1),
  204. ["^2 not less than or equal to 1$", "^oops$",
  205. "^2 not less than or equal to 1$",
  206. "^2 not less than or equal to 1 : oops$"])
  207. def testAssertGreater(self):
  208. self.assertMessages('assertGreater', (1, 2),
  209. ["^1 not greater than 2$", "^oops$",
  210. "^1 not greater than 2$",
  211. "^1 not greater than 2 : oops$"])
  212. def testAssertGreaterEqual(self):
  213. self.assertMessages('assertGreaterEqual', (1, 2),
  214. ["^1 not greater than or equal to 2$", "^oops$",
  215. "^1 not greater than or equal to 2$",
  216. "^1 not greater than or equal to 2 : oops$"])
  217. def testAssertIsNone(self):
  218. self.assertMessages('assertIsNone', ('not None',),
  219. ["^'not None' is not None$", "^oops$",
  220. "^'not None' is not None$",
  221. "^'not None' is not None : oops$"])
  222. def testAssertIsNotNone(self):
  223. self.assertMessages('assertIsNotNone', (None,),
  224. ["^unexpectedly None$", "^oops$",
  225. "^unexpectedly None$",
  226. "^unexpectedly None : oops$"])
  227. def testAssertIs(self):
  228. self.assertMessages('assertIs', (None, 'foo'),
  229. ["^None is not 'foo'$", "^oops$",
  230. "^None is not 'foo'$",
  231. "^None is not 'foo' : oops$"])
  232. def testAssertIsNot(self):
  233. self.assertMessages('assertIsNot', (None, None),
  234. ["^unexpectedly identical: None$", "^oops$",
  235. "^unexpectedly identical: None$",
  236. "^unexpectedly identical: None : oops$"])
  237. if __name__ == '__main__':
  238. unittest.main()