| # Licensed under the Apache License: http://www.apache.org/licenses/LICENSE-2.0 | |
| # For details: https://bitbucket.org/ned/coveragepy/src/default/NOTICE.txt | |
| """Implementations of unittest features from the future.""" | |
| # Use unittest2 if it's available, otherwise unittest. This gives us | |
| # back-ported features for 2.6. | |
| try: | |
| import unittest2 as unittest | |
| except ImportError: | |
| import unittest | |
| def unittest_has(method): | |
| """Does `unittest.TestCase` have `method` defined?""" | |
| return hasattr(unittest.TestCase, method) | |
| class TestCase(unittest.TestCase): | |
| """Just like unittest.TestCase, but with assert methods added. | |
| Designed to be compatible with 3.1 unittest. Methods are only defined if | |
| `unittest` doesn't have them. | |
| """ | |
| # pylint: disable=missing-docstring | |
| # Many Pythons have this method defined. But PyPy3 has a bug with it | |
| # somehow (https://bitbucket.org/pypy/pypy/issues/2092), so always use our | |
| # own implementation that works everywhere, at least for the ways we're | |
| # calling it. | |
| def assertCountEqual(self, s1, s2): | |
| """Assert these have the same elements, regardless of order.""" | |
| self.assertEqual(sorted(s1), sorted(s2)) | |
| if not unittest_has('assertRaisesRegex'): | |
| def assertRaisesRegex(self, *args, **kwargs): | |
| return self.assertRaisesRegexp(*args, **kwargs) | |
| if not unittest_has('assertRegex'): | |
| def assertRegex(self, *args, **kwargs): | |
| return self.assertRegexpMatches(*args, **kwargs) |