mirror of
				https://github.com/python/cpython.git
				synced 2025-11-04 15:41:43 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			1843 lines
		
	
	
	
		
			71 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			1843 lines
		
	
	
	
		
			71 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
import contextlib
 | 
						|
import difflib
 | 
						|
import pprint
 | 
						|
import pickle
 | 
						|
import re
 | 
						|
import sys
 | 
						|
import logging
 | 
						|
import warnings
 | 
						|
import weakref
 | 
						|
import inspect
 | 
						|
 | 
						|
from copy import deepcopy
 | 
						|
from test import support
 | 
						|
 | 
						|
import unittest
 | 
						|
 | 
						|
from unittest.test.support import (
 | 
						|
    TestEquality, TestHashing, LoggingResult, LegacyLoggingResult,
 | 
						|
    ResultWithNoStartTestRunStopTestRun
 | 
						|
)
 | 
						|
from test.support import captured_stderr
 | 
						|
 | 
						|
 | 
						|
log_foo = logging.getLogger('foo')
 | 
						|
log_foobar = logging.getLogger('foo.bar')
 | 
						|
log_quux = logging.getLogger('quux')
 | 
						|
 | 
						|
 | 
						|
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 MyException()
 | 
						|
            def test(self): pass
 | 
						|
 | 
						|
        self.assertEqual(Test().id()[-13:], '.Test.runTest')
 | 
						|
 | 
						|
        # test that TestCase can be instantiated with no args
 | 
						|
        # primarily for use at the interactive interpreter
 | 
						|
        test = unittest.TestCase()
 | 
						|
        test.assertEqual(3, 3)
 | 
						|
        with test.assertRaises(test.failureException):
 | 
						|
            test.assertEqual(3, 2)
 | 
						|
 | 
						|
        with self.assertRaises(AttributeError):
 | 
						|
            test.run()
 | 
						|
 | 
						|
    # "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 MyException()
 | 
						|
            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 MyException()
 | 
						|
            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', 'tearDown',
 | 
						|
                    'addError', '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',
 | 
						|
                    'tearDown', 'addError', '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', 'tearDown',
 | 
						|
                    'addFailure', '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',
 | 
						|
                    'tearDown', 'addFailure', '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()
 | 
						|
 | 
						|
    def _check_call_order__subtests(self, result, events, expected_events):
 | 
						|
        class Foo(Test.LoggingTestCase):
 | 
						|
            def test(self):
 | 
						|
                super(Foo, self).test()
 | 
						|
                for i in [1, 2, 3]:
 | 
						|
                    with self.subTest(i=i):
 | 
						|
                        if i == 1:
 | 
						|
                            self.fail('failure')
 | 
						|
                        for j in [2, 3]:
 | 
						|
                            with self.subTest(j=j):
 | 
						|
                                if i * j == 6:
 | 
						|
                                    raise RuntimeError('raised by Foo.test')
 | 
						|
                1 / 0
 | 
						|
 | 
						|
        # Order is the following:
 | 
						|
        # i=1 => subtest failure
 | 
						|
        # i=2, j=2 => subtest success
 | 
						|
        # i=2, j=3 => subtest error
 | 
						|
        # i=3, j=2 => subtest error
 | 
						|
        # i=3, j=3 => subtest success
 | 
						|
        # toplevel => error
 | 
						|
        Foo(events).run(result)
 | 
						|
        self.assertEqual(events, expected_events)
 | 
						|
 | 
						|
    def test_run_call_order__subtests(self):
 | 
						|
        events = []
 | 
						|
        result = LoggingResult(events)
 | 
						|
        expected = ['startTest', 'setUp', 'test', 'tearDown',
 | 
						|
                    'addSubTestFailure', 'addSubTestSuccess',
 | 
						|
                    'addSubTestFailure', 'addSubTestFailure',
 | 
						|
                    'addSubTestSuccess', 'addError', 'stopTest']
 | 
						|
        self._check_call_order__subtests(result, events, expected)
 | 
						|
 | 
						|
    def test_run_call_order__subtests_legacy(self):
 | 
						|
        # With a legacy result object (without an addSubTest method),
 | 
						|
        # text execution stops after the first subtest failure.
 | 
						|
        events = []
 | 
						|
        result = LegacyLoggingResult(events)
 | 
						|
        expected = ['startTest', 'setUp', 'test', 'tearDown',
 | 
						|
                    'addFailure', 'stopTest']
 | 
						|
        self._check_call_order__subtests(result, events, expected)
 | 
						|
 | 
						|
    def _check_call_order__subtests_success(self, result, events, expected_events):
 | 
						|
        class Foo(Test.LoggingTestCase):
 | 
						|
            def test(self):
 | 
						|
                super(Foo, self).test()
 | 
						|
                for i in [1, 2]:
 | 
						|
                    with self.subTest(i=i):
 | 
						|
                        for j in [2, 3]:
 | 
						|
                            with self.subTest(j=j):
 | 
						|
                                pass
 | 
						|
 | 
						|
        Foo(events).run(result)
 | 
						|
        self.assertEqual(events, expected_events)
 | 
						|
 | 
						|
    def test_run_call_order__subtests_success(self):
 | 
						|
        events = []
 | 
						|
        result = LoggingResult(events)
 | 
						|
        # The 6 subtest successes are individually recorded, in addition
 | 
						|
        # to the whole test success.
 | 
						|
        expected = (['startTest', 'setUp', 'test', 'tearDown']
 | 
						|
                    + 6 * ['addSubTestSuccess']
 | 
						|
                    + ['addSuccess', 'stopTest'])
 | 
						|
        self._check_call_order__subtests_success(result, events, expected)
 | 
						|
 | 
						|
    def test_run_call_order__subtests_success_legacy(self):
 | 
						|
        # With a legacy result, only the whole test success is recorded.
 | 
						|
        events = []
 | 
						|
        result = LegacyLoggingResult(events)
 | 
						|
        expected = ['startTest', 'setUp', 'test', 'tearDown',
 | 
						|
                    'addSuccess', 'stopTest']
 | 
						|
        self._check_call_order__subtests_success(result, events, expected)
 | 
						|
 | 
						|
    def test_run_call_order__subtests_failfast(self):
 | 
						|
        events = []
 | 
						|
        result = LoggingResult(events)
 | 
						|
        result.failfast = True
 | 
						|
 | 
						|
        class Foo(Test.LoggingTestCase):
 | 
						|
            def test(self):
 | 
						|
                super(Foo, self).test()
 | 
						|
                with self.subTest(i=1):
 | 
						|
                    self.fail('failure')
 | 
						|
                with self.subTest(i=2):
 | 
						|
                    self.fail('failure')
 | 
						|
                self.fail('failure')
 | 
						|
 | 
						|
        expected = ['startTest', 'setUp', 'test', 'tearDown',
 | 
						|
                    'addSubTestFailure', 'stopTest']
 | 
						|
        Foo(events).run(result)
 | 
						|
        self.assertEqual(events, expected)
 | 
						|
 | 
						|
    def test_subtests_failfast(self):
 | 
						|
        # Ensure proper test flow with subtests and failfast (issue #22894)
 | 
						|
        events = []
 | 
						|
 | 
						|
        class Foo(unittest.TestCase):
 | 
						|
            def test_a(self):
 | 
						|
                with self.subTest():
 | 
						|
                    events.append('a1')
 | 
						|
                events.append('a2')
 | 
						|
 | 
						|
            def test_b(self):
 | 
						|
                with self.subTest():
 | 
						|
                    events.append('b1')
 | 
						|
                with self.subTest():
 | 
						|
                    self.fail('failure')
 | 
						|
                events.append('b2')
 | 
						|
 | 
						|
            def test_c(self):
 | 
						|
                events.append('c')
 | 
						|
 | 
						|
        result = unittest.TestResult()
 | 
						|
        result.failfast = True
 | 
						|
        suite = unittest.makeSuite(Foo)
 | 
						|
        suite.run(result)
 | 
						|
 | 
						|
        expected = ['a1', 'a2', 'b1']
 | 
						|
        self.assertEqual(events, expected)
 | 
						|
 | 
						|
    def test_subtests_debug(self):
 | 
						|
        # Test debug() with a test that uses subTest() (bpo-34900)
 | 
						|
        events = []
 | 
						|
 | 
						|
        class Foo(unittest.TestCase):
 | 
						|
            def test_a(self):
 | 
						|
                events.append('test case')
 | 
						|
                with self.subTest():
 | 
						|
                    events.append('subtest 1')
 | 
						|
 | 
						|
        Foo('test_a').debug()
 | 
						|
 | 
						|
        self.assertEqual(events, ['test case', 'subtest 1'])
 | 
						|
 | 
						|
    # "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(), str)
 | 
						|
 | 
						|
 | 
						|
    # "If result is omitted or None, a temporary result object is created,
 | 
						|
    # used, and is made available to the caller. As TestCase owns the
 | 
						|
    # temporary result startTestRun and stopTestRun are called.
 | 
						|
 | 
						|
    def test_run__uses_defaultTestResult(self):
 | 
						|
        events = []
 | 
						|
        defaultResult = LoggingResult(events)
 | 
						|
 | 
						|
        class Foo(unittest.TestCase):
 | 
						|
            def test(self):
 | 
						|
                events.append('test')
 | 
						|
 | 
						|
            def defaultTestResult(self):
 | 
						|
                return defaultResult
 | 
						|
 | 
						|
        # Make run() find a result object on its own
 | 
						|
        result = Foo('test').run()
 | 
						|
 | 
						|
        self.assertIs(result, defaultResult)
 | 
						|
        expected = ['startTestRun', 'startTest', 'test', 'addSuccess',
 | 
						|
            'stopTest', 'stopTestRun']
 | 
						|
        self.assertEqual(events, expected)
 | 
						|
 | 
						|
 | 
						|
    # "The result object is returned to run's caller"
 | 
						|
    def test_run__returns_given_result(self):
 | 
						|
 | 
						|
        class Foo(unittest.TestCase):
 | 
						|
            def test(self):
 | 
						|
                pass
 | 
						|
 | 
						|
        result = unittest.TestResult()
 | 
						|
 | 
						|
        retval = Foo('test').run(result)
 | 
						|
        self.assertIs(retval, result)
 | 
						|
 | 
						|
 | 
						|
    # "The same effect [as method run] may be had by simply calling the
 | 
						|
    # TestCase instance."
 | 
						|
    def test_call__invoking_an_instance_delegates_to_run(self):
 | 
						|
        resultIn = unittest.TestResult()
 | 
						|
        resultOut = unittest.TestResult()
 | 
						|
 | 
						|
        class Foo(unittest.TestCase):
 | 
						|
            def test(self):
 | 
						|
                pass
 | 
						|
 | 
						|
            def run(self, result):
 | 
						|
                self.assertIs(result, resultIn)
 | 
						|
                return resultOut
 | 
						|
 | 
						|
        retval = Foo('test')(resultIn)
 | 
						|
 | 
						|
        self.assertIs(retval, resultOut)
 | 
						|
 | 
						|
 | 
						|
    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.assertFalse(s1 == s2)
 | 
						|
        def AllSnakesCreatedEqual(a, b, msg=None):
 | 
						|
            return type(a) == type(b) == 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):
 | 
						|
        with warnings.catch_warnings():
 | 
						|
            warnings.simplefilter("ignore", DeprecationWarning)
 | 
						|
 | 
						|
            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})
 | 
						|
 | 
						|
            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': '\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**5
 | 
						|
        self.addCleanup(lambda: setattr(self, '_diffThreshold', old_threshold))
 | 
						|
 | 
						|
        # under the threshold: diff marker (^) in error message
 | 
						|
        s = 'x' * (2**4)
 | 
						|
        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 = 'x' * (2**6)
 | 
						|
        # 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 testAssertEqual_shorten(self):
 | 
						|
        # set a lower threshold value and add a cleanup to restore it
 | 
						|
        old_threshold = self._diffThreshold
 | 
						|
        self._diffThreshold = 0
 | 
						|
        self.addCleanup(lambda: setattr(self, '_diffThreshold', old_threshold))
 | 
						|
 | 
						|
        s = 'x' * 100
 | 
						|
        s1, s2 = s + 'a', s + 'b'
 | 
						|
        with self.assertRaises(self.failureException) as cm:
 | 
						|
            self.assertEqual(s1, s2)
 | 
						|
        c = 'xxxx[35 chars]' + 'x' * 61
 | 
						|
        self.assertEqual(str(cm.exception), "'%sa' != '%sb'" % (c, c))
 | 
						|
        self.assertEqual(s + 'a', s + 'a')
 | 
						|
 | 
						|
        p = 'y' * 50
 | 
						|
        s1, s2 = s + 'a' + p, s + 'b' + p
 | 
						|
        with self.assertRaises(self.failureException) as cm:
 | 
						|
            self.assertEqual(s1, s2)
 | 
						|
        c = 'xxxx[85 chars]xxxxxxxxxxx'
 | 
						|
        self.assertEqual(str(cm.exception), "'%sa%s' != '%sb%s'" % (c, p, c, p))
 | 
						|
 | 
						|
        p = 'y' * 100
 | 
						|
        s1, s2 = s + 'a' + p, s + 'b' + p
 | 
						|
        with self.assertRaises(self.failureException) as cm:
 | 
						|
            self.assertEqual(s1, s2)
 | 
						|
        c = 'xxxx[91 chars]xxxxx'
 | 
						|
        d = 'y' * 40 + '[56 chars]yyyy'
 | 
						|
        self.assertEqual(str(cm.exception), "'%sa%s' != '%sb%s'" % (c, d, c, d))
 | 
						|
 | 
						|
    def testAssertCountEqual(self):
 | 
						|
        a = object()
 | 
						|
        self.assertCountEqual([1, 2, 3], [3, 2, 1])
 | 
						|
        self.assertCountEqual(['foo', 'bar', 'baz'], ['bar', 'baz', 'foo'])
 | 
						|
        self.assertCountEqual([a, a, 2, 2, 3], (a, 2, 3, a, 2))
 | 
						|
        self.assertCountEqual([1, "2", "a", "a"], ["a", "2", True, "a"])
 | 
						|
        self.assertRaises(self.failureException, self.assertCountEqual,
 | 
						|
                          [1, 2] + [3] * 100, [1] * 100 + [2, 3])
 | 
						|
        self.assertRaises(self.failureException, self.assertCountEqual,
 | 
						|
                          [1, "2", "a", "a"], ["a", "2", True, 1])
 | 
						|
        self.assertRaises(self.failureException, self.assertCountEqual,
 | 
						|
                          [10], [10, 11])
 | 
						|
        self.assertRaises(self.failureException, self.assertCountEqual,
 | 
						|
                          [10, 11], [10])
 | 
						|
        self.assertRaises(self.failureException, self.assertCountEqual,
 | 
						|
                          [10, 11, 10], [10, 11])
 | 
						|
 | 
						|
        # Test that sequences of unhashable objects can be tested for sameness:
 | 
						|
        self.assertCountEqual([[1, 2], [3, 4], 0], [False, [3, 4], [1, 2]])
 | 
						|
        # Test that iterator of unhashable objects can be tested for sameness:
 | 
						|
        self.assertCountEqual(iter([1, 2, [], 3, 4]),
 | 
						|
                              iter([1, 2, [], 3, 4]))
 | 
						|
 | 
						|
        # hashable types, but not orderable
 | 
						|
        self.assertRaises(self.failureException, self.assertCountEqual,
 | 
						|
                          [], [divmod, 'x', 1, 5j, 2j, frozenset()])
 | 
						|
        # comparing dicts
 | 
						|
        self.assertCountEqual([{'a': 1}, {'b': 2}], [{'b': 2}, {'a': 1}])
 | 
						|
        # comparing heterogeneous non-hashable sequences
 | 
						|
        self.assertCountEqual([1, 'x', divmod, []], [divmod, [], 'x', 1])
 | 
						|
        self.assertRaises(self.failureException, self.assertCountEqual,
 | 
						|
                          [], [divmod, [], 'x', 1, 5j, 2j, set()])
 | 
						|
        self.assertRaises(self.failureException, self.assertCountEqual,
 | 
						|
                          [[1]], [[2]])
 | 
						|
 | 
						|
        # Same elements, but not same sequence length
 | 
						|
        self.assertRaises(self.failureException, self.assertCountEqual,
 | 
						|
                          [1, 1, 2], [2, 1])
 | 
						|
        self.assertRaises(self.failureException, self.assertCountEqual,
 | 
						|
                          [1, 1, "2", "a", "a"], ["2", "2", True, "a"])
 | 
						|
        self.assertRaises(self.failureException, self.assertCountEqual,
 | 
						|
                          [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.assertCountEqual(a, b)
 | 
						|
 | 
						|
        # test utility functions supporting assertCountEqual()
 | 
						|
 | 
						|
        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 bytes
 | 
						|
        self.assertGreater(b'bug', b'ant')
 | 
						|
        self.assertGreaterEqual(b'bug', b'ant')
 | 
						|
        self.assertGreaterEqual(b'ant', b'ant')
 | 
						|
        self.assertLess(b'ant', b'bug')
 | 
						|
        self.assertLessEqual(b'ant', b'bug')
 | 
						|
        self.assertLessEqual(b'ant', b'ant')
 | 
						|
        self.assertRaises(self.failureException, self.assertGreater, b'ant', b'bug')
 | 
						|
        self.assertRaises(self.failureException, self.assertGreater, b'ant', b'ant')
 | 
						|
        self.assertRaises(self.failureException, self.assertGreaterEqual, b'ant',
 | 
						|
                          b'bug')
 | 
						|
        self.assertRaises(self.failureException, self.assertLess, b'bug', b'ant')
 | 
						|
        self.assertRaises(self.failureException, self.assertLess, b'ant', b'ant')
 | 
						|
        self.assertRaises(self.failureException, self.assertLessEqual, b'bug', b'ant')
 | 
						|
 | 
						|
    def testAssertMultiLineEqual(self):
 | 
						|
        sample_text = """\
 | 
						|
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 = """\
 | 
						|
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 = """\
 | 
						|
- 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
 | 
						|
        try:
 | 
						|
            self.assertMultiLineEqual(sample_text, revised_sample_text)
 | 
						|
        except self.failureException as e:
 | 
						|
            # need to remove the first line of the error message
 | 
						|
            error = str(e).split('\n', 1)[1]
 | 
						|
            self.assertEqual(sample_text_error, error)
 | 
						|
 | 
						|
    def testAssertEqualSingleLine(self):
 | 
						|
        sample_text = "laden swallows fly slowly"
 | 
						|
        revised_sample_text = "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:
 | 
						|
            # need to remove the first line of the error message
 | 
						|
            error = str(e).split('\n', 1)[1]
 | 
						|
            self.assertEqual(sample_text_error, error)
 | 
						|
 | 
						|
    def testEqualityBytesWarning(self):
 | 
						|
        if sys.flags.bytes_warning:
 | 
						|
            def bytes_warning():
 | 
						|
                return self.assertWarnsRegex(BytesWarning,
 | 
						|
                            'Comparison between bytes and string')
 | 
						|
        else:
 | 
						|
            def bytes_warning():
 | 
						|
                return contextlib.ExitStack()
 | 
						|
 | 
						|
        with bytes_warning(), self.assertRaises(self.failureException):
 | 
						|
            self.assertEqual('a', b'a')
 | 
						|
        with bytes_warning():
 | 
						|
            self.assertNotEqual('a', b'a')
 | 
						|
 | 
						|
        a = [0, 'a']
 | 
						|
        b = [0, b'a']
 | 
						|
        with bytes_warning(), self.assertRaises(self.failureException):
 | 
						|
            self.assertListEqual(a, b)
 | 
						|
        with bytes_warning(), self.assertRaises(self.failureException):
 | 
						|
            self.assertTupleEqual(tuple(a), tuple(b))
 | 
						|
        with bytes_warning(), self.assertRaises(self.failureException):
 | 
						|
            self.assertSequenceEqual(a, tuple(b))
 | 
						|
        with bytes_warning(), self.assertRaises(self.failureException):
 | 
						|
            self.assertSequenceEqual(tuple(a), b)
 | 
						|
        with bytes_warning(), self.assertRaises(self.failureException):
 | 
						|
            self.assertSequenceEqual('a', b'a')
 | 
						|
        with bytes_warning(), self.assertRaises(self.failureException):
 | 
						|
            self.assertSetEqual(set(a), set(b))
 | 
						|
 | 
						|
        with self.assertRaises(self.failureException):
 | 
						|
            self.assertListEqual(a, tuple(b))
 | 
						|
        with self.assertRaises(self.failureException):
 | 
						|
            self.assertTupleEqual(tuple(a), b)
 | 
						|
 | 
						|
        a = [0, b'a']
 | 
						|
        b = [0]
 | 
						|
        with self.assertRaises(self.failureException):
 | 
						|
            self.assertListEqual(a, b)
 | 
						|
        with self.assertRaises(self.failureException):
 | 
						|
            self.assertTupleEqual(tuple(a), tuple(b))
 | 
						|
        with self.assertRaises(self.failureException):
 | 
						|
            self.assertSequenceEqual(a, tuple(b))
 | 
						|
        with self.assertRaises(self.failureException):
 | 
						|
            self.assertSequenceEqual(tuple(a), b)
 | 
						|
        with self.assertRaises(self.failureException):
 | 
						|
            self.assertSetEqual(set(a), set(b))
 | 
						|
 | 
						|
        a = [0]
 | 
						|
        b = [0, b'a']
 | 
						|
        with self.assertRaises(self.failureException):
 | 
						|
            self.assertListEqual(a, b)
 | 
						|
        with self.assertRaises(self.failureException):
 | 
						|
            self.assertTupleEqual(tuple(a), tuple(b))
 | 
						|
        with self.assertRaises(self.failureException):
 | 
						|
            self.assertSequenceEqual(a, tuple(b))
 | 
						|
        with self.assertRaises(self.failureException):
 | 
						|
            self.assertSequenceEqual(tuple(a), b)
 | 
						|
        with self.assertRaises(self.failureException):
 | 
						|
            self.assertSetEqual(set(a), set(b))
 | 
						|
 | 
						|
        with bytes_warning(), self.assertRaises(self.failureException):
 | 
						|
            self.assertDictEqual({'a': 0}, {b'a': 0})
 | 
						|
        with self.assertRaises(self.failureException):
 | 
						|
            self.assertDictEqual({}, {b'a': 0})
 | 
						|
        with self.assertRaises(self.failureException):
 | 
						|
            self.assertDictEqual({b'a': 0}, {})
 | 
						|
 | 
						|
        with self.assertRaises(self.failureException):
 | 
						|
            self.assertCountEqual([b'a', b'a'], [b'a', b'a', b'a'])
 | 
						|
        with bytes_warning():
 | 
						|
            self.assertCountEqual(['a', b'a'], ['a', b'a'])
 | 
						|
        with bytes_warning(), self.assertRaises(self.failureException):
 | 
						|
            self.assertCountEqual(['a', 'a'], [b'a', b'a'])
 | 
						|
        with bytes_warning(), self.assertRaises(self.failureException):
 | 
						|
            self.assertCountEqual(['a', 'a', []], [b'a', b'a', []])
 | 
						|
 | 
						|
    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 testAssertRegex(self):
 | 
						|
        self.assertRegex('asdfabasdf', r'ab+')
 | 
						|
        self.assertRaises(self.failureException, self.assertRegex,
 | 
						|
                          '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 the function is None
 | 
						|
        with self.assertRaises(TypeError):
 | 
						|
            self.assertRaises(ExceptionMock, None)
 | 
						|
        # 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
 | 
						|
        # Custom message
 | 
						|
        with self.assertRaisesRegex(self.failureException, 'foobar'):
 | 
						|
            with self.assertRaises(ExceptionMock, msg='foobar'):
 | 
						|
                pass
 | 
						|
        # Invalid keyword argument
 | 
						|
        with self.assertRaisesRegex(TypeError, 'foobar'):
 | 
						|
            with self.assertRaises(ExceptionMock, foobar=42):
 | 
						|
                pass
 | 
						|
        # Failure when another exception is raised
 | 
						|
        with self.assertRaises(ExceptionMock):
 | 
						|
            self.assertRaises(ValueError, Stub)
 | 
						|
 | 
						|
    def testAssertRaisesNoExceptionType(self):
 | 
						|
        with self.assertRaises(TypeError):
 | 
						|
            self.assertRaises()
 | 
						|
        with self.assertRaises(TypeError):
 | 
						|
            self.assertRaises(1)
 | 
						|
        with self.assertRaises(TypeError):
 | 
						|
            self.assertRaises(object)
 | 
						|
        with self.assertRaises(TypeError):
 | 
						|
            self.assertRaises((ValueError, 1))
 | 
						|
        with self.assertRaises(TypeError):
 | 
						|
            self.assertRaises((ValueError, object))
 | 
						|
 | 
						|
    def testAssertRaisesRefcount(self):
 | 
						|
        # bpo-23890: assertRaises() must not keep objects alive longer
 | 
						|
        # than expected
 | 
						|
        def func() :
 | 
						|
            try:
 | 
						|
                raise ValueError
 | 
						|
            except ValueError:
 | 
						|
                raise ValueError
 | 
						|
 | 
						|
        refcount = sys.getrefcount(func)
 | 
						|
        self.assertRaises(ValueError, func)
 | 
						|
        self.assertEqual(refcount, sys.getrefcount(func))
 | 
						|
 | 
						|
    def testAssertRaisesRegex(self):
 | 
						|
        class ExceptionMock(Exception):
 | 
						|
            pass
 | 
						|
 | 
						|
        def Stub():
 | 
						|
            raise ExceptionMock('We expect')
 | 
						|
 | 
						|
        self.assertRaisesRegex(ExceptionMock, re.compile('expect$'), Stub)
 | 
						|
        self.assertRaisesRegex(ExceptionMock, 'expect$', Stub)
 | 
						|
        with self.assertRaises(TypeError):
 | 
						|
            self.assertRaisesRegex(ExceptionMock, 'expect$', None)
 | 
						|
 | 
						|
    def testAssertNotRaisesRegex(self):
 | 
						|
        self.assertRaisesRegex(
 | 
						|
                self.failureException, '^Exception not raised by <lambda>$',
 | 
						|
                self.assertRaisesRegex, Exception, re.compile('x'),
 | 
						|
                lambda: None)
 | 
						|
        self.assertRaisesRegex(
 | 
						|
                self.failureException, '^Exception not raised by <lambda>$',
 | 
						|
                self.assertRaisesRegex, Exception, 'x',
 | 
						|
                lambda: None)
 | 
						|
        # Custom message
 | 
						|
        with self.assertRaisesRegex(self.failureException, 'foobar'):
 | 
						|
            with self.assertRaisesRegex(Exception, 'expect', msg='foobar'):
 | 
						|
                pass
 | 
						|
        # Invalid keyword argument
 | 
						|
        with self.assertRaisesRegex(TypeError, 'foobar'):
 | 
						|
            with self.assertRaisesRegex(Exception, 'expect', foobar=42):
 | 
						|
                pass
 | 
						|
 | 
						|
    def testAssertRaisesRegexInvalidRegex(self):
 | 
						|
        # Issue 20145.
 | 
						|
        class MyExc(Exception):
 | 
						|
            pass
 | 
						|
        self.assertRaises(TypeError, self.assertRaisesRegex, MyExc, lambda: True)
 | 
						|
 | 
						|
    def testAssertWarnsRegexInvalidRegex(self):
 | 
						|
        # Issue 20145.
 | 
						|
        class MyWarn(Warning):
 | 
						|
            pass
 | 
						|
        self.assertRaises(TypeError, self.assertWarnsRegex, MyWarn, lambda: True)
 | 
						|
 | 
						|
    def testAssertRaisesRegexMismatch(self):
 | 
						|
        def Stub():
 | 
						|
            raise Exception('Unexpected')
 | 
						|
 | 
						|
        self.assertRaisesRegex(
 | 
						|
                self.failureException,
 | 
						|
                r'"\^Expected\$" does not match "Unexpected"',
 | 
						|
                self.assertRaisesRegex, Exception, '^Expected$',
 | 
						|
                Stub)
 | 
						|
        self.assertRaisesRegex(
 | 
						|
                self.failureException,
 | 
						|
                r'"\^Expected\$" does not match "Unexpected"',
 | 
						|
                self.assertRaisesRegex, 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 testAssertRaisesRegexNoExceptionType(self):
 | 
						|
        with self.assertRaises(TypeError):
 | 
						|
            self.assertRaisesRegex()
 | 
						|
        with self.assertRaises(TypeError):
 | 
						|
            self.assertRaisesRegex(ValueError)
 | 
						|
        with self.assertRaises(TypeError):
 | 
						|
            self.assertRaisesRegex(1, 'expect')
 | 
						|
        with self.assertRaises(TypeError):
 | 
						|
            self.assertRaisesRegex(object, 'expect')
 | 
						|
        with self.assertRaises(TypeError):
 | 
						|
            self.assertRaisesRegex((ValueError, 1), 'expect')
 | 
						|
        with self.assertRaises(TypeError):
 | 
						|
            self.assertRaisesRegex((ValueError, object), 'expect')
 | 
						|
 | 
						|
    def testAssertWarnsCallable(self):
 | 
						|
        def _runtime_warn():
 | 
						|
            warnings.warn("foo", RuntimeWarning)
 | 
						|
        # Success when the right warning is triggered, even several times
 | 
						|
        self.assertWarns(RuntimeWarning, _runtime_warn)
 | 
						|
        self.assertWarns(RuntimeWarning, _runtime_warn)
 | 
						|
        # A tuple of warning classes is accepted
 | 
						|
        self.assertWarns((DeprecationWarning, RuntimeWarning), _runtime_warn)
 | 
						|
        # *args and **kwargs also work
 | 
						|
        self.assertWarns(RuntimeWarning,
 | 
						|
                         warnings.warn, "foo", category=RuntimeWarning)
 | 
						|
        # Failure when no warning is triggered
 | 
						|
        with self.assertRaises(self.failureException):
 | 
						|
            self.assertWarns(RuntimeWarning, lambda: 0)
 | 
						|
        # Failure when the function is None
 | 
						|
        with self.assertRaises(TypeError):
 | 
						|
            self.assertWarns(RuntimeWarning, None)
 | 
						|
        # Failure when another warning is triggered
 | 
						|
        with warnings.catch_warnings():
 | 
						|
            # Force default filter (in case tests are run with -We)
 | 
						|
            warnings.simplefilter("default", RuntimeWarning)
 | 
						|
            with self.assertRaises(self.failureException):
 | 
						|
                self.assertWarns(DeprecationWarning, _runtime_warn)
 | 
						|
        # Filters for other warnings are not modified
 | 
						|
        with warnings.catch_warnings():
 | 
						|
            warnings.simplefilter("error", RuntimeWarning)
 | 
						|
            with self.assertRaises(RuntimeWarning):
 | 
						|
                self.assertWarns(DeprecationWarning, _runtime_warn)
 | 
						|
 | 
						|
    def testAssertWarnsContext(self):
 | 
						|
        # Believe it or not, it is preferable to duplicate all tests above,
 | 
						|
        # to make sure the __warningregistry__ $@ is circumvented correctly.
 | 
						|
        def _runtime_warn():
 | 
						|
            warnings.warn("foo", RuntimeWarning)
 | 
						|
        _runtime_warn_lineno = inspect.getsourcelines(_runtime_warn)[1]
 | 
						|
        with self.assertWarns(RuntimeWarning) as cm:
 | 
						|
            _runtime_warn()
 | 
						|
        # A tuple of warning classes is accepted
 | 
						|
        with self.assertWarns((DeprecationWarning, RuntimeWarning)) as cm:
 | 
						|
            _runtime_warn()
 | 
						|
        # The context manager exposes various useful attributes
 | 
						|
        self.assertIsInstance(cm.warning, RuntimeWarning)
 | 
						|
        self.assertEqual(cm.warning.args[0], "foo")
 | 
						|
        self.assertIn("test_case.py", cm.filename)
 | 
						|
        self.assertEqual(cm.lineno, _runtime_warn_lineno + 1)
 | 
						|
        # Same with several warnings
 | 
						|
        with self.assertWarns(RuntimeWarning):
 | 
						|
            _runtime_warn()
 | 
						|
            _runtime_warn()
 | 
						|
        with self.assertWarns(RuntimeWarning):
 | 
						|
            warnings.warn("foo", category=RuntimeWarning)
 | 
						|
        # Failure when no warning is triggered
 | 
						|
        with self.assertRaises(self.failureException):
 | 
						|
            with self.assertWarns(RuntimeWarning):
 | 
						|
                pass
 | 
						|
        # Custom message
 | 
						|
        with self.assertRaisesRegex(self.failureException, 'foobar'):
 | 
						|
            with self.assertWarns(RuntimeWarning, msg='foobar'):
 | 
						|
                pass
 | 
						|
        # Invalid keyword argument
 | 
						|
        with self.assertRaisesRegex(TypeError, 'foobar'):
 | 
						|
            with self.assertWarns(RuntimeWarning, foobar=42):
 | 
						|
                pass
 | 
						|
        # Failure when another warning is triggered
 | 
						|
        with warnings.catch_warnings():
 | 
						|
            # Force default filter (in case tests are run with -We)
 | 
						|
            warnings.simplefilter("default", RuntimeWarning)
 | 
						|
            with self.assertRaises(self.failureException):
 | 
						|
                with self.assertWarns(DeprecationWarning):
 | 
						|
                    _runtime_warn()
 | 
						|
        # Filters for other warnings are not modified
 | 
						|
        with warnings.catch_warnings():
 | 
						|
            warnings.simplefilter("error", RuntimeWarning)
 | 
						|
            with self.assertRaises(RuntimeWarning):
 | 
						|
                with self.assertWarns(DeprecationWarning):
 | 
						|
                    _runtime_warn()
 | 
						|
 | 
						|
    def testAssertWarnsNoExceptionType(self):
 | 
						|
        with self.assertRaises(TypeError):
 | 
						|
            self.assertWarns()
 | 
						|
        with self.assertRaises(TypeError):
 | 
						|
            self.assertWarns(1)
 | 
						|
        with self.assertRaises(TypeError):
 | 
						|
            self.assertWarns(object)
 | 
						|
        with self.assertRaises(TypeError):
 | 
						|
            self.assertWarns((UserWarning, 1))
 | 
						|
        with self.assertRaises(TypeError):
 | 
						|
            self.assertWarns((UserWarning, object))
 | 
						|
        with self.assertRaises(TypeError):
 | 
						|
            self.assertWarns((UserWarning, Exception))
 | 
						|
 | 
						|
    def testAssertWarnsRegexCallable(self):
 | 
						|
        def _runtime_warn(msg):
 | 
						|
            warnings.warn(msg, RuntimeWarning)
 | 
						|
        self.assertWarnsRegex(RuntimeWarning, "o+",
 | 
						|
                              _runtime_warn, "foox")
 | 
						|
        # Failure when no warning is triggered
 | 
						|
        with self.assertRaises(self.failureException):
 | 
						|
            self.assertWarnsRegex(RuntimeWarning, "o+",
 | 
						|
                                  lambda: 0)
 | 
						|
        # Failure when the function is None
 | 
						|
        with self.assertRaises(TypeError):
 | 
						|
            self.assertWarnsRegex(RuntimeWarning, "o+", None)
 | 
						|
        # Failure when another warning is triggered
 | 
						|
        with warnings.catch_warnings():
 | 
						|
            # Force default filter (in case tests are run with -We)
 | 
						|
            warnings.simplefilter("default", RuntimeWarning)
 | 
						|
            with self.assertRaises(self.failureException):
 | 
						|
                self.assertWarnsRegex(DeprecationWarning, "o+",
 | 
						|
                                      _runtime_warn, "foox")
 | 
						|
        # Failure when message doesn't match
 | 
						|
        with self.assertRaises(self.failureException):
 | 
						|
            self.assertWarnsRegex(RuntimeWarning, "o+",
 | 
						|
                                  _runtime_warn, "barz")
 | 
						|
        # A little trickier: we ask RuntimeWarnings to be raised, and then
 | 
						|
        # check for some of them.  It is implementation-defined whether
 | 
						|
        # non-matching RuntimeWarnings are simply re-raised, or produce a
 | 
						|
        # failureException.
 | 
						|
        with warnings.catch_warnings():
 | 
						|
            warnings.simplefilter("error", RuntimeWarning)
 | 
						|
            with self.assertRaises((RuntimeWarning, self.failureException)):
 | 
						|
                self.assertWarnsRegex(RuntimeWarning, "o+",
 | 
						|
                                      _runtime_warn, "barz")
 | 
						|
 | 
						|
    def testAssertWarnsRegexContext(self):
 | 
						|
        # Same as above, but with assertWarnsRegex as a context manager
 | 
						|
        def _runtime_warn(msg):
 | 
						|
            warnings.warn(msg, RuntimeWarning)
 | 
						|
        _runtime_warn_lineno = inspect.getsourcelines(_runtime_warn)[1]
 | 
						|
        with self.assertWarnsRegex(RuntimeWarning, "o+") as cm:
 | 
						|
            _runtime_warn("foox")
 | 
						|
        self.assertIsInstance(cm.warning, RuntimeWarning)
 | 
						|
        self.assertEqual(cm.warning.args[0], "foox")
 | 
						|
        self.assertIn("test_case.py", cm.filename)
 | 
						|
        self.assertEqual(cm.lineno, _runtime_warn_lineno + 1)
 | 
						|
        # Failure when no warning is triggered
 | 
						|
        with self.assertRaises(self.failureException):
 | 
						|
            with self.assertWarnsRegex(RuntimeWarning, "o+"):
 | 
						|
                pass
 | 
						|
        # Custom message
 | 
						|
        with self.assertRaisesRegex(self.failureException, 'foobar'):
 | 
						|
            with self.assertWarnsRegex(RuntimeWarning, 'o+', msg='foobar'):
 | 
						|
                pass
 | 
						|
        # Invalid keyword argument
 | 
						|
        with self.assertRaisesRegex(TypeError, 'foobar'):
 | 
						|
            with self.assertWarnsRegex(RuntimeWarning, 'o+', foobar=42):
 | 
						|
                pass
 | 
						|
        # Failure when another warning is triggered
 | 
						|
        with warnings.catch_warnings():
 | 
						|
            # Force default filter (in case tests are run with -We)
 | 
						|
            warnings.simplefilter("default", RuntimeWarning)
 | 
						|
            with self.assertRaises(self.failureException):
 | 
						|
                with self.assertWarnsRegex(DeprecationWarning, "o+"):
 | 
						|
                    _runtime_warn("foox")
 | 
						|
        # Failure when message doesn't match
 | 
						|
        with self.assertRaises(self.failureException):
 | 
						|
            with self.assertWarnsRegex(RuntimeWarning, "o+"):
 | 
						|
                _runtime_warn("barz")
 | 
						|
        # A little trickier: we ask RuntimeWarnings to be raised, and then
 | 
						|
        # check for some of them.  It is implementation-defined whether
 | 
						|
        # non-matching RuntimeWarnings are simply re-raised, or produce a
 | 
						|
        # failureException.
 | 
						|
        with warnings.catch_warnings():
 | 
						|
            warnings.simplefilter("error", RuntimeWarning)
 | 
						|
            with self.assertRaises((RuntimeWarning, self.failureException)):
 | 
						|
                with self.assertWarnsRegex(RuntimeWarning, "o+"):
 | 
						|
                    _runtime_warn("barz")
 | 
						|
 | 
						|
    def testAssertWarnsRegexNoExceptionType(self):
 | 
						|
        with self.assertRaises(TypeError):
 | 
						|
            self.assertWarnsRegex()
 | 
						|
        with self.assertRaises(TypeError):
 | 
						|
            self.assertWarnsRegex(UserWarning)
 | 
						|
        with self.assertRaises(TypeError):
 | 
						|
            self.assertWarnsRegex(1, 'expect')
 | 
						|
        with self.assertRaises(TypeError):
 | 
						|
            self.assertWarnsRegex(object, 'expect')
 | 
						|
        with self.assertRaises(TypeError):
 | 
						|
            self.assertWarnsRegex((UserWarning, 1), 'expect')
 | 
						|
        with self.assertRaises(TypeError):
 | 
						|
            self.assertWarnsRegex((UserWarning, object), 'expect')
 | 
						|
        with self.assertRaises(TypeError):
 | 
						|
            self.assertWarnsRegex((UserWarning, Exception), 'expect')
 | 
						|
 | 
						|
    @contextlib.contextmanager
 | 
						|
    def assertNoStderr(self):
 | 
						|
        with captured_stderr() as buf:
 | 
						|
            yield
 | 
						|
        self.assertEqual(buf.getvalue(), "")
 | 
						|
 | 
						|
    def assertLogRecords(self, records, matches):
 | 
						|
        self.assertEqual(len(records), len(matches))
 | 
						|
        for rec, match in zip(records, matches):
 | 
						|
            self.assertIsInstance(rec, logging.LogRecord)
 | 
						|
            for k, v in match.items():
 | 
						|
                self.assertEqual(getattr(rec, k), v)
 | 
						|
 | 
						|
    def testAssertLogsDefaults(self):
 | 
						|
        # defaults: root logger, level INFO
 | 
						|
        with self.assertNoStderr():
 | 
						|
            with self.assertLogs() as cm:
 | 
						|
                log_foo.info("1")
 | 
						|
                log_foobar.debug("2")
 | 
						|
            self.assertEqual(cm.output, ["INFO:foo:1"])
 | 
						|
            self.assertLogRecords(cm.records, [{'name': 'foo'}])
 | 
						|
 | 
						|
    def testAssertLogsTwoMatchingMessages(self):
 | 
						|
        # Same, but with two matching log messages
 | 
						|
        with self.assertNoStderr():
 | 
						|
            with self.assertLogs() as cm:
 | 
						|
                log_foo.info("1")
 | 
						|
                log_foobar.debug("2")
 | 
						|
                log_quux.warning("3")
 | 
						|
            self.assertEqual(cm.output, ["INFO:foo:1", "WARNING:quux:3"])
 | 
						|
            self.assertLogRecords(cm.records,
 | 
						|
                                   [{'name': 'foo'}, {'name': 'quux'}])
 | 
						|
 | 
						|
    def checkAssertLogsPerLevel(self, level):
 | 
						|
        # Check level filtering
 | 
						|
        with self.assertNoStderr():
 | 
						|
            with self.assertLogs(level=level) as cm:
 | 
						|
                log_foo.warning("1")
 | 
						|
                log_foobar.error("2")
 | 
						|
                log_quux.critical("3")
 | 
						|
            self.assertEqual(cm.output, ["ERROR:foo.bar:2", "CRITICAL:quux:3"])
 | 
						|
            self.assertLogRecords(cm.records,
 | 
						|
                                   [{'name': 'foo.bar'}, {'name': 'quux'}])
 | 
						|
 | 
						|
    def testAssertLogsPerLevel(self):
 | 
						|
        self.checkAssertLogsPerLevel(logging.ERROR)
 | 
						|
        self.checkAssertLogsPerLevel('ERROR')
 | 
						|
 | 
						|
    def checkAssertLogsPerLogger(self, logger):
 | 
						|
        # Check per-logger filtering
 | 
						|
        with self.assertNoStderr():
 | 
						|
            with self.assertLogs(level='DEBUG') as outer_cm:
 | 
						|
                with self.assertLogs(logger, level='DEBUG') as cm:
 | 
						|
                    log_foo.info("1")
 | 
						|
                    log_foobar.debug("2")
 | 
						|
                    log_quux.warning("3")
 | 
						|
                self.assertEqual(cm.output, ["INFO:foo:1", "DEBUG:foo.bar:2"])
 | 
						|
                self.assertLogRecords(cm.records,
 | 
						|
                                       [{'name': 'foo'}, {'name': 'foo.bar'}])
 | 
						|
            # The outer catchall caught the quux log
 | 
						|
            self.assertEqual(outer_cm.output, ["WARNING:quux:3"])
 | 
						|
 | 
						|
    def testAssertLogsPerLogger(self):
 | 
						|
        self.checkAssertLogsPerLogger(logging.getLogger('foo'))
 | 
						|
        self.checkAssertLogsPerLogger('foo')
 | 
						|
 | 
						|
    def testAssertLogsFailureNoLogs(self):
 | 
						|
        # Failure due to no logs
 | 
						|
        with self.assertNoStderr():
 | 
						|
            with self.assertRaises(self.failureException):
 | 
						|
                with self.assertLogs():
 | 
						|
                    pass
 | 
						|
 | 
						|
    def testAssertLogsFailureLevelTooHigh(self):
 | 
						|
        # Failure due to level too high
 | 
						|
        with self.assertNoStderr():
 | 
						|
            with self.assertRaises(self.failureException):
 | 
						|
                with self.assertLogs(level='WARNING'):
 | 
						|
                    log_foo.info("1")
 | 
						|
 | 
						|
    def testAssertLogsFailureMismatchingLogger(self):
 | 
						|
        # Failure due to mismatching logger (and the logged message is
 | 
						|
        # passed through)
 | 
						|
        with self.assertLogs('quux', level='ERROR'):
 | 
						|
            with self.assertRaises(self.failureException):
 | 
						|
                with self.assertLogs('foo'):
 | 
						|
                    log_quux.error("1")
 | 
						|
 | 
						|
    def testDeprecatedMethodNames(self):
 | 
						|
        """
 | 
						|
        Test that the deprecated methods raise a DeprecationWarning. See #9424.
 | 
						|
        """
 | 
						|
        old = (
 | 
						|
            (self.failIfEqual, (3, 5)),
 | 
						|
            (self.assertNotEquals, (3, 5)),
 | 
						|
            (self.failUnlessEqual, (3, 3)),
 | 
						|
            (self.assertEquals, (3, 3)),
 | 
						|
            (self.failUnlessAlmostEqual, (2.0, 2.0)),
 | 
						|
            (self.assertAlmostEquals, (2.0, 2.0)),
 | 
						|
            (self.failIfAlmostEqual, (3.0, 5.0)),
 | 
						|
            (self.assertNotAlmostEquals, (3.0, 5.0)),
 | 
						|
            (self.failUnless, (True,)),
 | 
						|
            (self.assert_, (True,)),
 | 
						|
            (self.failUnlessRaises, (TypeError, lambda _: 3.14 + 'spam')),
 | 
						|
            (self.failIf, (False,)),
 | 
						|
            (self.assertDictContainsSubset, (dict(a=1, b=2), dict(a=1, b=2, c=3))),
 | 
						|
            (self.assertRaisesRegexp, (KeyError, 'foo', lambda: {}['foo'])),
 | 
						|
            (self.assertRegexpMatches, ('bar', 'bar')),
 | 
						|
        )
 | 
						|
        for meth, args in old:
 | 
						|
            with self.assertWarns(DeprecationWarning):
 | 
						|
                meth(*args)
 | 
						|
 | 
						|
    # disable this test for now. When the version where the fail* methods will
 | 
						|
    # be removed is decided, re-enable it and update the version
 | 
						|
    def _testDeprecatedFailMethods(self):
 | 
						|
        """Test that the deprecated fail* methods get removed in 3.x"""
 | 
						|
        if sys.version_info[:2] < (3, 3):
 | 
						|
            return
 | 
						|
        deprecated_names = [
 | 
						|
            'failIfEqual', 'failUnlessEqual', 'failUnlessAlmostEqual',
 | 
						|
            'failIfAlmostEqual', 'failUnless', 'failUnlessRaises', 'failIf',
 | 
						|
            'assertDictContainsSubset',
 | 
						|
        ]
 | 
						|
        for deprecated_name in deprecated_names:
 | 
						|
            with self.assertRaises(AttributeError):
 | 
						|
                getattr(self, deprecated_name)  # remove these in 3.x
 | 
						|
 | 
						|
    def testDeepcopy(self):
 | 
						|
        # Issue: 5660
 | 
						|
        class TestableTest(unittest.TestCase):
 | 
						|
            def testNothing(self):
 | 
						|
                pass
 | 
						|
 | 
						|
        test = TestableTest('testNothing')
 | 
						|
 | 
						|
        # This shouldn't blow up
 | 
						|
        deepcopy(test)
 | 
						|
 | 
						|
    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)
 | 
						|
 | 
						|
            # exercise the TestCase instance in a way that will invoke
 | 
						|
            # the type equality lookup mechanism
 | 
						|
            unpickled_test.assertEqual(set(), set())
 | 
						|
 | 
						|
    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 testSkippingEverywhere(self):
 | 
						|
        def _skip(self=None):
 | 
						|
            raise unittest.SkipTest('some reason')
 | 
						|
        def nothing(self):
 | 
						|
            pass
 | 
						|
 | 
						|
        class Test1(unittest.TestCase):
 | 
						|
            test_something = _skip
 | 
						|
 | 
						|
        class Test2(unittest.TestCase):
 | 
						|
            setUp = _skip
 | 
						|
            test_something = nothing
 | 
						|
 | 
						|
        class Test3(unittest.TestCase):
 | 
						|
            test_something = nothing
 | 
						|
            tearDown = _skip
 | 
						|
 | 
						|
        class Test4(unittest.TestCase):
 | 
						|
            def test_something(self):
 | 
						|
                self.addCleanup(_skip)
 | 
						|
 | 
						|
        for klass in (Test1, Test2, Test3, Test4):
 | 
						|
            result = unittest.TestResult()
 | 
						|
            klass('test_something').run(result)
 | 
						|
            self.assertEqual(len(result.skipped), 1)
 | 
						|
            self.assertEqual(result.testsRun, 1)
 | 
						|
 | 
						|
    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)
 | 
						|
 | 
						|
    @support.cpython_only
 | 
						|
    def testNoCycles(self):
 | 
						|
        case = unittest.TestCase()
 | 
						|
        wr = weakref.ref(case)
 | 
						|
        with support.disable_gc():
 | 
						|
            del case
 | 
						|
            self.assertFalse(wr())
 | 
						|
 | 
						|
    def test_no_exception_leak(self):
 | 
						|
        # Issue #19880: TestCase.run() should not keep a reference
 | 
						|
        # to the exception
 | 
						|
        class MyException(Exception):
 | 
						|
            ninstance = 0
 | 
						|
 | 
						|
            def __init__(self):
 | 
						|
                MyException.ninstance += 1
 | 
						|
                Exception.__init__(self)
 | 
						|
 | 
						|
            def __del__(self):
 | 
						|
                MyException.ninstance -= 1
 | 
						|
 | 
						|
        class TestCase(unittest.TestCase):
 | 
						|
            def test1(self):
 | 
						|
                raise MyException()
 | 
						|
 | 
						|
            @unittest.expectedFailure
 | 
						|
            def test2(self):
 | 
						|
                raise MyException()
 | 
						|
 | 
						|
        for method_name in ('test1', 'test2'):
 | 
						|
            testcase = TestCase(method_name)
 | 
						|
            testcase.run()
 | 
						|
            self.assertEqual(MyException.ninstance, 0)
 | 
						|
 | 
						|
 | 
						|
if __name__ == "__main__":
 | 
						|
    unittest.main()
 |