diff options
Diffstat (limited to 'tests/test_summary_class.py')
| -rw-r--r-- | tests/test_summary_class.py | 26 |
1 files changed, 20 insertions, 6 deletions
diff --git a/tests/test_summary_class.py b/tests/test_summary_class.py index 2a8cd7c6..bc4466bc 100644 --- a/tests/test_summary_class.py +++ b/tests/test_summary_class.py @@ -7,10 +7,14 @@ import collections import unittest import os.path -try: - from cStringIO import StringIO -except ImportError: - from StringIO import StringIO +import sys +if sys.version_info < (3, ): + try: + from cStringIO import StringIO + except ImportError: + from StringIO import StringIO +else: + from io import StringIO from coverage import summary, data, control, config LINES_1 = { @@ -35,6 +39,10 @@ class TestSummaryReporterConfiguration(unittest.TestCase): printer.report([], destination) return destination.getvalue() + if sys.version_info < (2, 7): + def assertNotIn(self, needle, haystack): + self.assertTrue(needle not in haystack) + def test_defaults(self): """Run the report with no configuration options.""" data = self.get_coverage_data() @@ -49,7 +57,10 @@ class TestSummaryReporterConfiguration(unittest.TestCase): opts = config.CoverageConfig() opts.from_args(show_missing=True) report = self.get_summary_text(data, opts) - self.assertIn('Missing', report) + if sys.version_info > (2, 7): + self.assertIn('Missing', report) + else: + self.assertTrue('Missing' in report) self.assertNotIn('Branch', report) def test_sort_report(self): @@ -62,4 +73,7 @@ class TestSummaryReporterConfiguration(unittest.TestCase): filename = os.path.splitext(os.path.basename(__file__))[0] location1 = report.find('helpers') location2 = report.find(filename) - self.assertLess(location1, location2) + if sys.version_info > (2, 7): + self.assertLess(location1, location2) + else: + self.assertTrue(location1 < location2) |