1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171 |
- import difflib
- import pprint
- import pickle
- import re
- import sys
- from copy import deepcopy
- from test import test_support
- import unittest
- from unittest.test.support import (
- TestEquality, TestHashing, LoggingResult, ResultWithNoStartTestRunStopTestRun
- )
- class Test(object):
- "Keep these TestCase classes out of the main namespace"
- class Foo(unittest.TestCase):
- def runTest(self): pass
- def test1(self): pass
- class Bar(Foo):
- def test2(self): pass
- class LoggingTestCase(unittest.TestCase):
- """A test case which logs its calls."""
- def __init__(self, events):
- super(Test.LoggingTestCase, self).__init__('test')
- self.events = events
- def setUp(self):
- self.events.append('setUp')
- def test(self):
- self.events.append('test')
- def tearDown(self):
- self.events.append('tearDown')
- class Test_TestCase(unittest.TestCase, TestEquality, TestHashing):
- ### Set up attributes used by inherited tests
- ################################################################
- # Used by TestHashing.test_hash and TestEquality.test_eq
- eq_pairs = [(Test.Foo('test1'), Test.Foo('test1'))]
- # Used by TestEquality.test_ne
- ne_pairs = [(Test.Foo('test1'), Test.Foo('runTest'))
- ,(Test.Foo('test1'), Test.Bar('test1'))
- ,(Test.Foo('test1'), Test.Bar('test2'))]
- ################################################################
- ### /Set up attributes used by inherited tests
- # "class TestCase([methodName])"
- # ...
- # "Each instance of TestCase will run a single test method: the
- # method named methodName."
- # ...
- # "methodName defaults to "runTest"."
- #
- # Make sure it really is optional, and that it defaults to the proper
- # thing.
- def test_init__no_test_name(self):
- class Test(unittest.TestCase):
- def runTest(self): raise TypeError()
- def test(self): pass
- self.assertEqual(Test().id()[-13:], '.Test.runTest')
- # "class TestCase([methodName])"
- # ...
- # "Each instance of TestCase will run a single test method: the
- # method named methodName."
- def test_init__test_name__valid(self):
- class Test(unittest.TestCase):
- def runTest(self): raise TypeError()
- def test(self): pass
- self.assertEqual(Test('test').id()[-10:], '.Test.test')
- # "class TestCase([methodName])"
- # ...
- # "Each instance of TestCase will run a single test method: the
- # method named methodName."
- def test_init__test_name__invalid(self):
- class Test(unittest.TestCase):
- def runTest(self): raise TypeError()
- def test(self): pass
- try:
- Test('testfoo')
- except ValueError:
- pass
- else:
- self.fail("Failed to raise ValueError")
- # "Return the number of tests represented by the this test object. For
- # TestCase instances, this will always be 1"
- def test_countTestCases(self):
- class Foo(unittest.TestCase):
- def test(self): pass
- self.assertEqual(Foo('test').countTestCases(), 1)
- # "Return the default type of test result object to be used to run this
- # test. For TestCase instances, this will always be
- # unittest.TestResult; subclasses of TestCase should
- # override this as necessary."
- def test_defaultTestResult(self):
- class Foo(unittest.TestCase):
- def runTest(self):
- pass
- result = Foo().defaultTestResult()
- self.assertEqual(type(result), unittest.TestResult)
- # "When a setUp() method is defined, the test runner will run that method
- # prior to each test. Likewise, if a tearDown() method is defined, the
- # test runner will invoke that method after each test. In the example,
- # setUp() was used to create a fresh sequence for each test."
- #
- # Make sure the proper call order is maintained, even if setUp() raises
- # an exception.
- def test_run_call_order__error_in_setUp(self):
- events = []
- result = LoggingResult(events)
- class Foo(Test.LoggingTestCase):
- def setUp(self):
- super(Foo, self).setUp()
- raise RuntimeError('raised by Foo.setUp')
- Foo(events).run(result)
- expected = ['startTest', 'setUp', 'addError', 'stopTest']
- self.assertEqual(events, expected)
- # "With a temporary result stopTestRun is called when setUp errors.
- def test_run_call_order__error_in_setUp_default_result(self):
- events = []
- class Foo(Test.LoggingTestCase):
- def defaultTestResult(self):
- return LoggingResult(self.events)
- def setUp(self):
- super(Foo, self).setUp()
- raise RuntimeError('raised by Foo.setUp')
- Foo(events).run()
- expected = ['startTestRun', 'startTest', 'setUp', 'addError',
- 'stopTest', 'stopTestRun']
- self.assertEqual(events, expected)
- # "When a setUp() method is defined, the test runner will run that method
- # prior to each test. Likewise, if a tearDown() method is defined, the
- # test runner will invoke that method after each test. In the example,
- # setUp() was used to create a fresh sequence for each test."
- #
- # Make sure the proper call order is maintained, even if the test raises
- # an error (as opposed to a failure).
- def test_run_call_order__error_in_test(self):
- events = []
- result = LoggingResult(events)
- class Foo(Test.LoggingTestCase):
- def test(self):
- super(Foo, self).test()
- raise RuntimeError('raised by Foo.test')
- expected = ['startTest', 'setUp', 'test', 'addError', 'tearDown',
- 'stopTest']
- Foo(events).run(result)
- self.assertEqual(events, expected)
- # "With a default result, an error in the test still results in stopTestRun
- # being called."
- def test_run_call_order__error_in_test_default_result(self):
- events = []
- class Foo(Test.LoggingTestCase):
- def defaultTestResult(self):
- return LoggingResult(self.events)
- def test(self):
- super(Foo, self).test()
- raise RuntimeError('raised by Foo.test')
- expected = ['startTestRun', 'startTest', 'setUp', 'test', 'addError',
- 'tearDown', 'stopTest', 'stopTestRun']
- Foo(events).run()
- self.assertEqual(events, expected)
- # "When a setUp() method is defined, the test runner will run that method
- # prior to each test. Likewise, if a tearDown() method is defined, the
- # test runner will invoke that method after each test. In the example,
- # setUp() was used to create a fresh sequence for each test."
- #
- # Make sure the proper call order is maintained, even if the test signals
- # a failure (as opposed to an error).
- def test_run_call_order__failure_in_test(self):
- events = []
- result = LoggingResult(events)
- class Foo(Test.LoggingTestCase):
- def test(self):
- super(Foo, self).test()
- self.fail('raised by Foo.test')
- expected = ['startTest', 'setUp', 'test', 'addFailure', 'tearDown',
- 'stopTest']
- Foo(events).run(result)
- self.assertEqual(events, expected)
- # "When a test fails with a default result stopTestRun is still called."
- def test_run_call_order__failure_in_test_default_result(self):
- class Foo(Test.LoggingTestCase):
- def defaultTestResult(self):
- return LoggingResult(self.events)
- def test(self):
- super(Foo, self).test()
- self.fail('raised by Foo.test')
- expected = ['startTestRun', 'startTest', 'setUp', 'test', 'addFailure',
- 'tearDown', 'stopTest', 'stopTestRun']
- events = []
- Foo(events).run()
- self.assertEqual(events, expected)
- # "When a setUp() method is defined, the test runner will run that method
- # prior to each test. Likewise, if a tearDown() method is defined, the
- # test runner will invoke that method after each test. In the example,
- # setUp() was used to create a fresh sequence for each test."
- #
- # Make sure the proper call order is maintained, even if tearDown() raises
- # an exception.
- def test_run_call_order__error_in_tearDown(self):
- events = []
- result = LoggingResult(events)
- class Foo(Test.LoggingTestCase):
- def tearDown(self):
- super(Foo, self).tearDown()
- raise RuntimeError('raised by Foo.tearDown')
- Foo(events).run(result)
- expected = ['startTest', 'setUp', 'test', 'tearDown', 'addError',
- 'stopTest']
- self.assertEqual(events, expected)
- # "When tearDown errors with a default result stopTestRun is still called."
- def test_run_call_order__error_in_tearDown_default_result(self):
- class Foo(Test.LoggingTestCase):
- def defaultTestResult(self):
- return LoggingResult(self.events)
- def tearDown(self):
- super(Foo, self).tearDown()
- raise RuntimeError('raised by Foo.tearDown')
- events = []
- Foo(events).run()
- expected = ['startTestRun', 'startTest', 'setUp', 'test', 'tearDown',
- 'addError', 'stopTest', 'stopTestRun']
- self.assertEqual(events, expected)
- # "TestCase.run() still works when the defaultTestResult is a TestResult
- # that does not support startTestRun and stopTestRun.
- def test_run_call_order_default_result(self):
- class Foo(unittest.TestCase):
- def defaultTestResult(self):
- return ResultWithNoStartTestRunStopTestRun()
- def test(self):
- pass
- Foo('test').run()
- # "This class attribute gives the exception raised by the test() method.
- # If a test framework needs to use a specialized exception, possibly to
- # carry additional information, it must subclass this exception in
- # order to ``play fair'' with the framework. The initial value of this
- # attribute is AssertionError"
- def test_failureException__default(self):
- class Foo(unittest.TestCase):
- def test(self):
- pass
- self.assertIs(Foo('test').failureException, AssertionError)
- # "This class attribute gives the exception raised by the test() method.
- # If a test framework needs to use a specialized exception, possibly to
- # carry additional information, it must subclass this exception in
- # order to ``play fair'' with the framework."
- #
- # Make sure TestCase.run() respects the designated failureException
- def test_failureException__subclassing__explicit_raise(self):
- events = []
- result = LoggingResult(events)
- class Foo(unittest.TestCase):
- def test(self):
- raise RuntimeError()
- failureException = RuntimeError
- self.assertIs(Foo('test').failureException, RuntimeError)
- Foo('test').run(result)
- expected = ['startTest', 'addFailure', 'stopTest']
- self.assertEqual(events, expected)
- # "This class attribute gives the exception raised by the test() method.
- # If a test framework needs to use a specialized exception, possibly to
- # carry additional information, it must subclass this exception in
- # order to ``play fair'' with the framework."
- #
- # Make sure TestCase.run() respects the designated failureException
- def test_failureException__subclassing__implicit_raise(self):
- events = []
- result = LoggingResult(events)
- class Foo(unittest.TestCase):
- def test(self):
- self.fail("foo")
- failureException = RuntimeError
- self.assertIs(Foo('test').failureException, RuntimeError)
- Foo('test').run(result)
- expected = ['startTest', 'addFailure', 'stopTest']
- self.assertEqual(events, expected)
- # "The default implementation does nothing."
- def test_setUp(self):
- class Foo(unittest.TestCase):
- def runTest(self):
- pass
- # ... and nothing should happen
- Foo().setUp()
- # "The default implementation does nothing."
- def test_tearDown(self):
- class Foo(unittest.TestCase):
- def runTest(self):
- pass
- # ... and nothing should happen
- Foo().tearDown()
- # "Return a string identifying the specific test case."
- #
- # Because of the vague nature of the docs, I'm not going to lock this
- # test down too much. Really all that can be asserted is that the id()
- # will be a string (either 8-byte or unicode -- again, because the docs
- # just say "string")
- def test_id(self):
- class Foo(unittest.TestCase):
- def runTest(self):
- pass
- self.assertIsInstance(Foo().id(), basestring)
- # "If result is omitted or None, a temporary result object is created
- # and used, but is not made available to the caller. As TestCase owns the
- # temporary result startTestRun and stopTestRun are called.
- def test_run__uses_defaultTestResult(self):
- events = []
- class Foo(unittest.TestCase):
- def test(self):
- events.append('test')
- def defaultTestResult(self):
- return LoggingResult(events)
- # Make run() find a result object on its own
- Foo('test').run()
- expected = ['startTestRun', 'startTest', 'test', 'addSuccess',
- 'stopTest', 'stopTestRun']
- self.assertEqual(events, expected)
- def testShortDescriptionWithoutDocstring(self):
- self.assertIsNone(self.shortDescription())
- @unittest.skipIf(sys.flags.optimize >= 2,
- "Docstrings are omitted with -O2 and above")
- def testShortDescriptionWithOneLineDocstring(self):
- """Tests shortDescription() for a method with a docstring."""
- self.assertEqual(
- self.shortDescription(),
- 'Tests shortDescription() for a method with a docstring.')
- @unittest.skipIf(sys.flags.optimize >= 2,
- "Docstrings are omitted with -O2 and above")
- def testShortDescriptionWithMultiLineDocstring(self):
- """Tests shortDescription() for a method with a longer docstring.
- This method ensures that only the first line of a docstring is
- returned used in the short description, no matter how long the
- whole thing is.
- """
- self.assertEqual(
- self.shortDescription(),
- 'Tests shortDescription() for a method with a longer '
- 'docstring.')
- def testAddTypeEqualityFunc(self):
- class SadSnake(object):
- """Dummy class for test_addTypeEqualityFunc."""
- s1, s2 = SadSnake(), SadSnake()
- self.assertNotEqual(s1, s2)
- def AllSnakesCreatedEqual(a, b, msg=None):
- return type(a) is type(b) is SadSnake
- self.addTypeEqualityFunc(SadSnake, AllSnakesCreatedEqual)
- self.assertEqual(s1, s2)
- # No this doesn't clean up and remove the SadSnake equality func
- # from this TestCase instance but since its a local nothing else
- # will ever notice that.
- def testAssertIs(self):
- thing = object()
- self.assertIs(thing, thing)
- self.assertRaises(self.failureException, self.assertIs, thing, object())
- def testAssertIsNot(self):
- thing = object()
- self.assertIsNot(thing, object())
- self.assertRaises(self.failureException, self.assertIsNot, thing, thing)
- def testAssertIsInstance(self):
- thing = []
- self.assertIsInstance(thing, list)
- self.assertRaises(self.failureException, self.assertIsInstance,
- thing, dict)
- def testAssertNotIsInstance(self):
- thing = []
- self.assertNotIsInstance(thing, dict)
- self.assertRaises(self.failureException, self.assertNotIsInstance,
- thing, list)
- def testAssertIn(self):
- animals = {'monkey': 'banana', 'cow': 'grass', 'seal': 'fish'}
- self.assertIn('a', 'abc')
- self.assertIn(2, [1, 2, 3])
- self.assertIn('monkey', animals)
- self.assertNotIn('d', 'abc')
- self.assertNotIn(0, [1, 2, 3])
- self.assertNotIn('otter', animals)
- self.assertRaises(self.failureException, self.assertIn, 'x', 'abc')
- self.assertRaises(self.failureException, self.assertIn, 4, [1, 2, 3])
- self.assertRaises(self.failureException, self.assertIn, 'elephant',
- animals)
- self.assertRaises(self.failureException, self.assertNotIn, 'c', 'abc')
- self.assertRaises(self.failureException, self.assertNotIn, 1, [1, 2, 3])
- self.assertRaises(self.failureException, self.assertNotIn, 'cow',
- animals)
- def testAssertDictContainsSubset(self):
- self.assertDictContainsSubset({}, {})
- self.assertDictContainsSubset({}, {'a': 1})
- self.assertDictContainsSubset({'a': 1}, {'a': 1})
- self.assertDictContainsSubset({'a': 1}, {'a': 1, 'b': 2})
- self.assertDictContainsSubset({'a': 1, 'b': 2}, {'a': 1, 'b': 2})
- with self.assertRaises(self.failureException):
- self.assertDictContainsSubset({1: "one"}, {})
- with self.assertRaises(self.failureException):
- self.assertDictContainsSubset({'a': 2}, {'a': 1})
- with self.assertRaises(self.failureException):
- self.assertDictContainsSubset({'c': 1}, {'a': 1})
- with self.assertRaises(self.failureException):
- self.assertDictContainsSubset({'a': 1, 'c': 1}, {'a': 1})
- with self.assertRaises(self.failureException):
- self.assertDictContainsSubset({'a': 1, 'c': 1}, {'a': 1})
- with test_support.check_warnings(("", UnicodeWarning)):
- one = ''.join(chr(i) for i in range(255))
- # this used to cause a UnicodeDecodeError constructing the failure msg
- with self.assertRaises(self.failureException):
- self.assertDictContainsSubset({'foo': one}, {'foo': u'\uFFFD'})
- def testAssertEqual(self):
- equal_pairs = [
- ((), ()),
- ({}, {}),
- ([], []),
- (set(), set()),
- (frozenset(), frozenset())]
- for a, b in equal_pairs:
- # This mess of try excepts is to test the assertEqual behavior
- # itself.
- try:
- self.assertEqual(a, b)
- except self.failureException:
- self.fail('assertEqual(%r, %r) failed' % (a, b))
- try:
- self.assertEqual(a, b, msg='foo')
- except self.failureException:
- self.fail('assertEqual(%r, %r) with msg= failed' % (a, b))
- try:
- self.assertEqual(a, b, 'foo')
- except self.failureException:
- self.fail('assertEqual(%r, %r) with third parameter failed' %
- (a, b))
- unequal_pairs = [
- ((), []),
- ({}, set()),
- (set([4,1]), frozenset([4,2])),
- (frozenset([4,5]), set([2,3])),
- (set([3,4]), set([5,4]))]
- for a, b in unequal_pairs:
- self.assertRaises(self.failureException, self.assertEqual, a, b)
- self.assertRaises(self.failureException, self.assertEqual, a, b,
- 'foo')
- self.assertRaises(self.failureException, self.assertEqual, a, b,
- msg='foo')
- def testEquality(self):
- self.assertListEqual([], [])
- self.assertTupleEqual((), ())
- self.assertSequenceEqual([], ())
- a = [0, 'a', []]
- b = []
- self.assertRaises(unittest.TestCase.failureException,
- self.assertListEqual, a, b)
- self.assertRaises(unittest.TestCase.failureException,
- self.assertListEqual, tuple(a), tuple(b))
- self.assertRaises(unittest.TestCase.failureException,
- self.assertSequenceEqual, a, tuple(b))
- b.extend(a)
- self.assertListEqual(a, b)
- self.assertTupleEqual(tuple(a), tuple(b))
- self.assertSequenceEqual(a, tuple(b))
- self.assertSequenceEqual(tuple(a), b)
- self.assertRaises(self.failureException, self.assertListEqual,
- a, tuple(b))
- self.assertRaises(self.failureException, self.assertTupleEqual,
- tuple(a), b)
- self.assertRaises(self.failureException, self.assertListEqual, None, b)
- self.assertRaises(self.failureException, self.assertTupleEqual, None,
- tuple(b))
- self.assertRaises(self.failureException, self.assertSequenceEqual,
- None, tuple(b))
- self.assertRaises(self.failureException, self.assertListEqual, 1, 1)
- self.assertRaises(self.failureException, self.assertTupleEqual, 1, 1)
- self.assertRaises(self.failureException, self.assertSequenceEqual,
- 1, 1)
- self.assertDictEqual({}, {})
- c = { 'x': 1 }
- d = {}
- self.assertRaises(unittest.TestCase.failureException,
- self.assertDictEqual, c, d)
- d.update(c)
- self.assertDictEqual(c, d)
- d['x'] = 0
- self.assertRaises(unittest.TestCase.failureException,
- self.assertDictEqual, c, d, 'These are unequal')
- self.assertRaises(self.failureException, self.assertDictEqual, None, d)
- self.assertRaises(self.failureException, self.assertDictEqual, [], d)
- self.assertRaises(self.failureException, self.assertDictEqual, 1, 1)
- def testAssertSequenceEqualMaxDiff(self):
- self.assertEqual(self.maxDiff, 80*8)
- seq1 = 'a' + 'x' * 80**2
- seq2 = 'b' + 'x' * 80**2
- diff = '\n'.join(difflib.ndiff(pprint.pformat(seq1).splitlines(),
- pprint.pformat(seq2).splitlines()))
- # the +1 is the leading \n added by assertSequenceEqual
- omitted = unittest.case.DIFF_OMITTED % (len(diff) + 1,)
- self.maxDiff = len(diff)//2
- try:
- self.assertSequenceEqual(seq1, seq2)
- except self.failureException as e:
- msg = e.args[0]
- else:
- self.fail('assertSequenceEqual did not fail.')
- self.assertLess(len(msg), len(diff))
- self.assertIn(omitted, msg)
- self.maxDiff = len(diff) * 2
- try:
- self.assertSequenceEqual(seq1, seq2)
- except self.failureException as e:
- msg = e.args[0]
- else:
- self.fail('assertSequenceEqual did not fail.')
- self.assertGreater(len(msg), len(diff))
- self.assertNotIn(omitted, msg)
- self.maxDiff = None
- try:
- self.assertSequenceEqual(seq1, seq2)
- except self.failureException as e:
- msg = e.args[0]
- else:
- self.fail('assertSequenceEqual did not fail.')
- self.assertGreater(len(msg), len(diff))
- self.assertNotIn(omitted, msg)
- def testTruncateMessage(self):
- self.maxDiff = 1
- message = self._truncateMessage('foo', 'bar')
- omitted = unittest.case.DIFF_OMITTED % len('bar')
- self.assertEqual(message, 'foo' + omitted)
- self.maxDiff = None
- message = self._truncateMessage('foo', 'bar')
- self.assertEqual(message, 'foobar')
- self.maxDiff = 4
- message = self._truncateMessage('foo', 'bar')
- self.assertEqual(message, 'foobar')
- def testAssertDictEqualTruncates(self):
- test = unittest.TestCase('assertEqual')
- def truncate(msg, diff):
- return 'foo'
- test._truncateMessage = truncate
- try:
- test.assertDictEqual({}, {1: 0})
- except self.failureException as e:
- self.assertEqual(str(e), 'foo')
- else:
- self.fail('assertDictEqual did not fail')
- def testAssertMultiLineEqualTruncates(self):
- test = unittest.TestCase('assertEqual')
- def truncate(msg, diff):
- return 'foo'
- test._truncateMessage = truncate
- try:
- test.assertMultiLineEqual('foo', 'bar')
- except self.failureException as e:
- self.assertEqual(str(e), 'foo')
- else:
- self.fail('assertMultiLineEqual did not fail')
- def testAssertEqual_diffThreshold(self):
- # check threshold value
- self.assertEqual(self._diffThreshold, 2**16)
- # disable madDiff to get diff markers
- self.maxDiff = None
- # set a lower threshold value and add a cleanup to restore it
- old_threshold = self._diffThreshold
- self._diffThreshold = 2**8
- self.addCleanup(lambda: setattr(self, '_diffThreshold', old_threshold))
- # under the threshold: diff marker (^) in error message
- s = u'x' * (2**7)
- with self.assertRaises(self.failureException) as cm:
- self.assertEqual(s + 'a', s + 'b')
- self.assertIn('^', str(cm.exception))
- self.assertEqual(s + 'a', s + 'a')
- # over the threshold: diff not used and marker (^) not in error message
- s = u'x' * (2**9)
- # if the path that uses difflib is taken, _truncateMessage will be
- # called -- replace it with explodingTruncation to verify that this
- # doesn't happen
- def explodingTruncation(message, diff):
- raise SystemError('this should not be raised')
- old_truncate = self._truncateMessage
- self._truncateMessage = explodingTruncation
- self.addCleanup(lambda: setattr(self, '_truncateMessage', old_truncate))
- s1, s2 = s + 'a', s + 'b'
- with self.assertRaises(self.failureException) as cm:
- self.assertEqual(s1, s2)
- self.assertNotIn('^', str(cm.exception))
- self.assertEqual(str(cm.exception), '%r != %r' % (s1, s2))
- self.assertEqual(s + 'a', s + 'a')
- def testAssertItemsEqual(self):
- a = object()
- self.assertItemsEqual([1, 2, 3], [3, 2, 1])
- self.assertItemsEqual(['foo', 'bar', 'baz'], ['bar', 'baz', 'foo'])
- self.assertItemsEqual([a, a, 2, 2, 3], (a, 2, 3, a, 2))
- self.assertItemsEqual([1, "2", "a", "a"], ["a", "2", True, "a"])
- self.assertRaises(self.failureException, self.assertItemsEqual,
- [1, 2] + [3] * 100, [1] * 100 + [2, 3])
- self.assertRaises(self.failureException, self.assertItemsEqual,
- [1, "2", "a", "a"], ["a", "2", True, 1])
- self.assertRaises(self.failureException, self.assertItemsEqual,
- [10], [10, 11])
- self.assertRaises(self.failureException, self.assertItemsEqual,
- [10, 11], [10])
- self.assertRaises(self.failureException, self.assertItemsEqual,
- [10, 11, 10], [10, 11])
- # Test that sequences of unhashable objects can be tested for sameness:
- self.assertItemsEqual([[1, 2], [3, 4], 0], [False, [3, 4], [1, 2]])
- # Test that iterator of unhashable objects can be tested for sameness:
- self.assertItemsEqual(iter([1, 2, [], 3, 4]),
- iter([1, 2, [], 3, 4]))
- # hashable types, but not orderable
- self.assertRaises(self.failureException, self.assertItemsEqual,
- [], [divmod, 'x', 1, 5j, 2j, frozenset()])
- # comparing dicts
- self.assertItemsEqual([{'a': 1}, {'b': 2}], [{'b': 2}, {'a': 1}])
- # comparing heterogenous non-hashable sequences
- self.assertItemsEqual([1, 'x', divmod, []], [divmod, [], 'x', 1])
- self.assertRaises(self.failureException, self.assertItemsEqual,
- [], [divmod, [], 'x', 1, 5j, 2j, set()])
- self.assertRaises(self.failureException, self.assertItemsEqual,
- [[1]], [[2]])
- # Same elements, but not same sequence length
- self.assertRaises(self.failureException, self.assertItemsEqual,
- [1, 1, 2], [2, 1])
- self.assertRaises(self.failureException, self.assertItemsEqual,
- [1, 1, "2", "a", "a"], ["2", "2", True, "a"])
- self.assertRaises(self.failureException, self.assertItemsEqual,
- [1, {'b': 2}, None, True], [{'b': 2}, True, None])
- # Same elements which don't reliably compare, in
- # different order, see issue 10242
- a = [{2,4}, {1,2}]
- b = a[::-1]
- self.assertItemsEqual(a, b)
- # test utility functions supporting assertItemsEqual()
- diffs = set(unittest.util._count_diff_all_purpose('aaabccd', 'abbbcce'))
- expected = {(3,1,'a'), (1,3,'b'), (1,0,'d'), (0,1,'e')}
- self.assertEqual(diffs, expected)
- diffs = unittest.util._count_diff_all_purpose([[]], [])
- self.assertEqual(diffs, [(1, 0, [])])
- diffs = set(unittest.util._count_diff_hashable('aaabccd', 'abbbcce'))
- expected = {(3,1,'a'), (1,3,'b'), (1,0,'d'), (0,1,'e')}
- self.assertEqual(diffs, expected)
- def testAssertSetEqual(self):
- set1 = set()
- set2 = set()
- self.assertSetEqual(set1, set2)
- self.assertRaises(self.failureException, self.assertSetEqual, None, set2)
- self.assertRaises(self.failureException, self.assertSetEqual, [], set2)
- self.assertRaises(self.failureException, self.assertSetEqual, set1, None)
- self.assertRaises(self.failureException, self.assertSetEqual, set1, [])
- set1 = set(['a'])
- set2 = set()
- self.assertRaises(self.failureException, self.assertSetEqual, set1, set2)
- set1 = set(['a'])
- set2 = set(['a'])
- self.assertSetEqual(set1, set2)
- set1 = set(['a'])
- set2 = set(['a', 'b'])
- self.assertRaises(self.failureException, self.assertSetEqual, set1, set2)
- set1 = set(['a'])
- set2 = frozenset(['a', 'b'])
- self.assertRaises(self.failureException, self.assertSetEqual, set1, set2)
- set1 = set(['a', 'b'])
- set2 = frozenset(['a', 'b'])
- self.assertSetEqual(set1, set2)
- set1 = set()
- set2 = "foo"
- self.assertRaises(self.failureException, self.assertSetEqual, set1, set2)
- self.assertRaises(self.failureException, self.assertSetEqual, set2, set1)
- # make sure any string formatting is tuple-safe
- set1 = set([(0, 1), (2, 3)])
- set2 = set([(4, 5)])
- self.assertRaises(self.failureException, self.assertSetEqual, set1, set2)
- def testInequality(self):
- # Try ints
- self.assertGreater(2, 1)
- self.assertGreaterEqual(2, 1)
- self.assertGreaterEqual(1, 1)
- self.assertLess(1, 2)
- self.assertLessEqual(1, 2)
- self.assertLessEqual(1, 1)
- self.assertRaises(self.failureException, self.assertGreater, 1, 2)
- self.assertRaises(self.failureException, self.assertGreater, 1, 1)
- self.assertRaises(self.failureException, self.assertGreaterEqual, 1, 2)
- self.assertRaises(self.failureException, self.assertLess, 2, 1)
- self.assertRaises(self.failureException, self.assertLess, 1, 1)
- self.assertRaises(self.failureException, self.assertLessEqual, 2, 1)
- # Try Floats
- self.assertGreater(1.1, 1.0)
- self.assertGreaterEqual(1.1, 1.0)
- self.assertGreaterEqual(1.0, 1.0)
- self.assertLess(1.0, 1.1)
- self.assertLessEqual(1.0, 1.1)
- self.assertLessEqual(1.0, 1.0)
- self.assertRaises(self.failureException, self.assertGreater, 1.0, 1.1)
- self.assertRaises(self.failureException, self.assertGreater, 1.0, 1.0)
- self.assertRaises(self.failureException, self.assertGreaterEqual, 1.0, 1.1)
- self.assertRaises(self.failureException, self.assertLess, 1.1, 1.0)
- self.assertRaises(self.failureException, self.assertLess, 1.0, 1.0)
- self.assertRaises(self.failureException, self.assertLessEqual, 1.1, 1.0)
- # Try Strings
- self.assertGreater('bug', 'ant')
- self.assertGreaterEqual('bug', 'ant')
- self.assertGreaterEqual('ant', 'ant')
- self.assertLess('ant', 'bug')
- self.assertLessEqual('ant', 'bug')
- self.assertLessEqual('ant', 'ant')
- self.assertRaises(self.failureException, self.assertGreater, 'ant', 'bug')
- self.assertRaises(self.failureException, self.assertGreater, 'ant', 'ant')
- self.assertRaises(self.failureException, self.assertGreaterEqual, 'ant', 'bug')
- self.assertRaises(self.failureException, self.assertLess, 'bug', 'ant')
- self.assertRaises(self.failureException, self.assertLess, 'ant', 'ant')
- self.assertRaises(self.failureException, self.assertLessEqual, 'bug', 'ant')
- # Try Unicode
- self.assertGreater(u'bug', u'ant')
- self.assertGreaterEqual(u'bug', u'ant')
- self.assertGreaterEqual(u'ant', u'ant')
- self.assertLess(u'ant', u'bug')
- self.assertLessEqual(u'ant', u'bug')
- self.assertLessEqual(u'ant', u'ant')
- self.assertRaises(self.failureException, self.assertGreater, u'ant', u'bug')
- self.assertRaises(self.failureException, self.assertGreater, u'ant', u'ant')
- self.assertRaises(self.failureException, self.assertGreaterEqual, u'ant',
- u'bug')
- self.assertRaises(self.failureException, self.assertLess, u'bug', u'ant')
- self.assertRaises(self.failureException, self.assertLess, u'ant', u'ant')
- self.assertRaises(self.failureException, self.assertLessEqual, u'bug', u'ant')
- # Try Mixed String/Unicode
- self.assertGreater('bug', u'ant')
- self.assertGreater(u'bug', 'ant')
- self.assertGreaterEqual('bug', u'ant')
- self.assertGreaterEqual(u'bug', 'ant')
- self.assertGreaterEqual('ant', u'ant')
- self.assertGreaterEqual(u'ant', 'ant')
- self.assertLess('ant', u'bug')
- self.assertLess(u'ant', 'bug')
- self.assertLessEqual('ant', u'bug')
- self.assertLessEqual(u'ant', 'bug')
- self.assertLessEqual('ant', u'ant')
- self.assertLessEqual(u'ant', 'ant')
- self.assertRaises(self.failureException, self.assertGreater, 'ant', u'bug')
- self.assertRaises(self.failureException, self.assertGreater, u'ant', 'bug')
- self.assertRaises(self.failureException, self.assertGreater, 'ant', u'ant')
- self.assertRaises(self.failureException, self.assertGreater, u'ant', 'ant')
- self.assertRaises(self.failureException, self.assertGreaterEqual, 'ant',
- u'bug')
- self.assertRaises(self.failureException, self.assertGreaterEqual, u'ant',
- 'bug')
- self.assertRaises(self.failureException, self.assertLess, 'bug', u'ant')
- self.assertRaises(self.failureException, self.assertLess, u'bug', 'ant')
- self.assertRaises(self.failureException, self.assertLess, 'ant', u'ant')
- self.assertRaises(self.failureException, self.assertLess, u'ant', 'ant')
- self.assertRaises(self.failureException, self.assertLessEqual, 'bug', u'ant')
- self.assertRaises(self.failureException, self.assertLessEqual, u'bug', 'ant')
- def testAssertMultiLineEqual(self):
- sample_text = b"""\
- http://www.python.org/doc/2.3/lib/module-unittest.html
- test case
- A test case is the smallest unit of testing. [...]
- """
- revised_sample_text = b"""\
- http://www.python.org/doc/2.4.1/lib/module-unittest.html
- test case
- A test case is the smallest unit of testing. [...] You may provide your
- own implementation that does not subclass from TestCase, of course.
- """
- sample_text_error = b"""\
- - http://www.python.org/doc/2.3/lib/module-unittest.html
- ? ^
- + http://www.python.org/doc/2.4.1/lib/module-unittest.html
- ? ^^^
- test case
- - A test case is the smallest unit of testing. [...]
- + A test case is the smallest unit of testing. [...] You may provide your
- ? +++++++++++++++++++++
- + own implementation that does not subclass from TestCase, of course.
- """
- self.maxDiff = None
- for type_changer in (lambda x: x, lambda x: x.decode('utf8')):
- try:
- self.assertMultiLineEqual(type_changer(sample_text),
- type_changer(revised_sample_text))
- except self.failureException, e:
- # need to remove the first line of the error message
- error = str(e).encode('utf8').split('\n', 1)[1]
- # assertMultiLineEqual is hooked up as the default for
- # unicode strings - so we can't use it for this check
- self.assertTrue(sample_text_error == error)
- def testAsertEqualSingleLine(self):
- sample_text = u"laden swallows fly slowly"
- revised_sample_text = u"unladen swallows fly quickly"
- sample_text_error = """\
- - laden swallows fly slowly
- ? ^^^^
- + unladen swallows fly quickly
- ? ++ ^^^^^
- """
- try:
- self.assertEqual(sample_text, revised_sample_text)
- except self.failureException as e:
- error = str(e).split('\n', 1)[1]
- self.assertTrue(sample_text_error == error)
- def testAssertIsNone(self):
- self.assertIsNone(None)
- self.assertRaises(self.failureException, self.assertIsNone, False)
- self.assertIsNotNone('DjZoPloGears on Rails')
- self.assertRaises(self.failureException, self.assertIsNotNone, None)
- def testAssertRegexpMatches(self):
- self.assertRegexpMatches('asdfabasdf', r'ab+')
- self.assertRaises(self.failureException, self.assertRegexpMatches,
- 'saaas', r'aaaa')
- def testAssertRaisesCallable(self):
- class ExceptionMock(Exception):
- pass
- def Stub():
- raise ExceptionMock('We expect')
- self.assertRaises(ExceptionMock, Stub)
- # A tuple of exception classes is accepted
- self.assertRaises((ValueError, ExceptionMock), Stub)
- # *args and **kwargs also work
- self.assertRaises(ValueError, int, '19', base=8)
- # Failure when no exception is raised
- with self.assertRaises(self.failureException):
- self.assertRaises(ExceptionMock, lambda: 0)
- # Failure when another exception is raised
- with self.assertRaises(ExceptionMock):
- self.assertRaises(ValueError, Stub)
- def testAssertRaisesContext(self):
- class ExceptionMock(Exception):
- pass
- def Stub():
- raise ExceptionMock('We expect')
- with self.assertRaises(ExceptionMock):
- Stub()
- # A tuple of exception classes is accepted
- with self.assertRaises((ValueError, ExceptionMock)) as cm:
- Stub()
- # The context manager exposes caught exception
- self.assertIsInstance(cm.exception, ExceptionMock)
- self.assertEqual(cm.exception.args[0], 'We expect')
- # *args and **kwargs also work
- with self.assertRaises(ValueError):
- int('19', base=8)
- # Failure when no exception is raised
- with self.assertRaises(self.failureException):
- with self.assertRaises(ExceptionMock):
- pass
- # Failure when another exception is raised
- with self.assertRaises(ExceptionMock):
- self.assertRaises(ValueError, Stub)
- def testAssertRaisesRegexp(self):
- class ExceptionMock(Exception):
- pass
- def Stub():
- raise ExceptionMock('We expect')
- self.assertRaisesRegexp(ExceptionMock, re.compile('expect$'), Stub)
- self.assertRaisesRegexp(ExceptionMock, 'expect$', Stub)
- self.assertRaisesRegexp(ExceptionMock, u'expect$', Stub)
- def testAssertNotRaisesRegexp(self):
- self.assertRaisesRegexp(
- self.failureException, '^Exception not raised$',
- self.assertRaisesRegexp, Exception, re.compile('x'),
- lambda: None)
- self.assertRaisesRegexp(
- self.failureException, '^Exception not raised$',
- self.assertRaisesRegexp, Exception, 'x',
- lambda: None)
- self.assertRaisesRegexp(
- self.failureException, '^Exception not raised$',
- self.assertRaisesRegexp, Exception, u'x',
- lambda: None)
- def testAssertRaisesRegexpInvalidRegexp(self):
- # Issue 20145.
- class MyExc(Exception):
- pass
- self.assertRaises(TypeError, self.assertRaisesRegexp, MyExc, lambda: True)
- def testAssertRaisesRegexpMismatch(self):
- def Stub():
- raise Exception('Unexpected')
- self.assertRaisesRegexp(
- self.failureException,
- r'"\^Expected\$" does not match "Unexpected"',
- self.assertRaisesRegexp, Exception, '^Expected$',
- Stub)
- self.assertRaisesRegexp(
- self.failureException,
- r'"\^Expected\$" does not match "Unexpected"',
- self.assertRaisesRegexp, Exception, u'^Expected$',
- Stub)
- self.assertRaisesRegexp(
- self.failureException,
- r'"\^Expected\$" does not match "Unexpected"',
- self.assertRaisesRegexp, Exception,
- re.compile('^Expected$'), Stub)
- def testAssertRaisesExcValue(self):
- class ExceptionMock(Exception):
- pass
- def Stub(foo):
- raise ExceptionMock(foo)
- v = "particular value"
- ctx = self.assertRaises(ExceptionMock)
- with ctx:
- Stub(v)
- e = ctx.exception
- self.assertIsInstance(e, ExceptionMock)
- self.assertEqual(e.args[0], v)
- def testSynonymAssertMethodNames(self):
- """Test undocumented method name synonyms.
- Please do not use these methods names in your own code.
- This test confirms their continued existence and functionality
- in order to avoid breaking existing code.
- """
- self.assertNotEquals(3, 5)
- self.assertEquals(3, 3)
- self.assertAlmostEquals(2.0, 2.0)
- self.assertNotAlmostEquals(3.0, 5.0)
- self.assert_(True)
- def testPendingDeprecationMethodNames(self):
- """Test fail* methods pending deprecation, they will warn in 3.2.
- Do not use these methods. They will go away in 3.3.
- """
- with test_support.check_warnings():
- self.failIfEqual(3, 5)
- self.failUnlessEqual(3, 3)
- self.failUnlessAlmostEqual(2.0, 2.0)
- self.failIfAlmostEqual(3.0, 5.0)
- self.failUnless(True)
- self.failUnlessRaises(TypeError, lambda _: 3.14 + u'spam')
- self.failIf(False)
- def testDeepcopy(self):
- # Issue: 5660
- class TestableTest(unittest.TestCase):
- def testNothing(self):
- pass
- test = TestableTest('testNothing')
- # This shouldn't blow up
- deepcopy(test)
- def testKeyboardInterrupt(self):
- def _raise(self=None):
- raise KeyboardInterrupt
- def nothing(self):
- pass
- class Test1(unittest.TestCase):
- test_something = _raise
- class Test2(unittest.TestCase):
- setUp = _raise
- test_something = nothing
- class Test3(unittest.TestCase):
- test_something = nothing
- tearDown = _raise
- class Test4(unittest.TestCase):
- def test_something(self):
- self.addCleanup(_raise)
- for klass in (Test1, Test2, Test3, Test4):
- with self.assertRaises(KeyboardInterrupt):
- klass('test_something').run()
- def testSystemExit(self):
- def _raise(self=None):
- raise SystemExit
- def nothing(self):
- pass
- class Test1(unittest.TestCase):
- test_something = _raise
- class Test2(unittest.TestCase):
- setUp = _raise
- test_something = nothing
- class Test3(unittest.TestCase):
- test_something = nothing
- tearDown = _raise
- class Test4(unittest.TestCase):
- def test_something(self):
- self.addCleanup(_raise)
- for klass in (Test1, Test2, Test3, Test4):
- result = unittest.TestResult()
- klass('test_something').run(result)
- self.assertEqual(len(result.errors), 1)
- self.assertEqual(result.testsRun, 1)
- def testPickle(self):
- # Issue 10326
- # Can't use TestCase classes defined in Test class as
- # pickle does not work with inner classes
- test = unittest.TestCase('run')
- for protocol in range(pickle.HIGHEST_PROTOCOL + 1):
- # blew up prior to fix
- pickled_test = pickle.dumps(test, protocol=protocol)
- unpickled_test = pickle.loads(pickled_test)
- self.assertEqual(test, unpickled_test)
- if __name__ == '__main__':
- unittest.main()
|