diff options
Diffstat (limited to 'tests/test_api.py')
| -rw-r--r-- | tests/test_api.py | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/tests/test_api.py b/tests/test_api.py index 107617ea..eb13896d 100644 --- a/tests/test_api.py +++ b/tests/test_api.py @@ -556,8 +556,10 @@ class ApiTest(CoverageTest): filenames = self.get_measured_filenames(data) suite_filename = filenames['testsuite.py'] - self.assertEqual([2, 8], data.lines(suite_filename, context="multiply_six")) - self.assertEqual([2, 5], data.lines(suite_filename, context="multiply_zero")) + self.assertEqual( + [2, 8], data.lines(suite_filename, contexts=["multiply_six"])) + self.assertEqual( + [2, 5], data.lines(suite_filename, contexts=["multiply_zero"])) def test_switch_context_with_static(self): # This test simulates a coverage-aware test runner, @@ -594,8 +596,10 @@ class ApiTest(CoverageTest): filenames = self.get_measured_filenames(data) suite_filename = filenames['testsuite.py'] - self.assertEqual([2, 8], data.lines(suite_filename, context="mysuite|multiply_six")) - self.assertEqual([2, 5], data.lines(suite_filename, context="mysuite|multiply_zero")) + self.assertEqual( + [2, 8], data.lines(suite_filename, contexts=["mysuite|multiply_six"])) + self.assertEqual( + [2, 5], data.lines(suite_filename, contexts=["mysuite|multiply_zero"])) def test_switch_context_unstarted(self): # Coverage must be started to switch context |