blob: 5a66f87358cff291615a849893d11bc4b83b67e7 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
|
"""Tests for Coverage.py's code parsing."""
import os, sys, textwrap
sys.path.insert(0, os.path.split(__file__)[0]) # Force relative import for Py3k
from coveragetest import CoverageTest
from coverage.parser import CodeParser
class ParserTest(CoverageTest):
"""Tests for Coverage.py's code parsing."""
def parse_source(self, text):
"""Parse `text` as source, and return the `CodeParser` used."""
text = textwrap.dedent(text)
cp = CodeParser(text)
cp.parse_source()
return cp
def test_exit_counts(self):
cp = self.parse_source("""\
# check some basic branch counting
class Foo:
def foo(self, a):
if a:
return 5
else:
return 7
class Bar:
pass
""")
self.assertEqual(cp.exit_counts(), {
2:1, 3:1, 4:2, 5:1, 7:1, 9:1, 10:1
})
|
