Move code to where it belongs

1 files changed, 14 insertions, 1 deletions

diff --git a/tests/test_results.py b/tests/test_results.py
index ec6301db..8acbcaec 100644
--- a/tests/test_results.py
+++ b/tests/test_results.py
@@ -6,7 +6,7 @@
 import pytest
 
 from coverage.misc import CoverageException
-from coverage.results import Numbers, should_fail_under
+from coverage.results import format_lines, Numbers, should_fail_under
 
 from tests.coveragetest import CoverageTest
 
@@ -111,3 +111,16 @@ def test_should_fail_under(total, fail_under, precision, result):
 def test_should_fail_under_invalid_value():
     with pytest.raises(CoverageException, match=r"fail_under=101"):
         should_fail_under(100.0, 101, 0)
+
+
+@pytest.mark.parametrize("statements, lines, result", [
+    (set([1,2,3,4,5,10,11,12,13,14]), set([1,2,5,10,11,13,14]), "1-2, 5-11, 13-14"),
+    ([1,2,3,4,5,10,11,12,13,14,98,99], [1,2,5,10,11,13,14,99], "1-2, 5-11, 13-14, 99"),
+    ([1,2,3,4,98,99,100,101,102,103,104], [1,2,99,102,103,104], "1-2, 99, 102-104"),
+    ([17], [17], "17"),
+    ([90,91,92,93,94,95], [90,91,92,93,94,95], "90-95"),
+    ([1, 2, 3, 4, 5], [], ""),
+    ([1, 2, 3, 4, 5], [4], "4"),
+])
+def test_format_lines(statements, lines, result):
+    assert format_lines(statements, lines) == result