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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
"""Tests for coverage.codeunit"""
import os, sys
from coverage.codeunit import code_unit_factory
from coverage.files import FileLocator
from coveragetest import CoverageTest
# pylint: disable-msg=F0401
# Unable to import 'aa' (No module named aa)
class CodeUnitTest(CoverageTest):
"""Tests for coverage.codeunit"""
def setUp(self):
super(CodeUnitTest, self).setUp()
# Parent class saves and restores sys.path, we can just modify it.
testmods = self.nice_file(os.path.dirname(__file__), 'modules')
sys.path.append(testmods)
def test_filenames(self):
acu = code_unit_factory("aa/afile.py", FileLocator())
bcu = code_unit_factory("aa/bb/bfile.py", FileLocator())
ccu = code_unit_factory("aa/bb/cc/cfile.py", FileLocator())
self.assertEqual(acu[0].name, "aa/afile")
self.assertEqual(bcu[0].name, "aa/bb/bfile")
self.assertEqual(ccu[0].name, "aa/bb/cc/cfile")
self.assertEqual(acu[0].flat_rootname(), "aa_afile")
self.assertEqual(bcu[0].flat_rootname(), "aa_bb_bfile")
self.assertEqual(ccu[0].flat_rootname(), "aa_bb_cc_cfile")
self.assertEqual(acu[0].source_file().read(), "# afile.py\n")
self.assertEqual(bcu[0].source_file().read(), "# bfile.py\n")
self.assertEqual(ccu[0].source_file().read(), "# cfile.py\n")
def test_modules(self):
import aa, aa.bb, aa.bb.cc
cu = code_unit_factory([aa, aa.bb, aa.bb.cc], FileLocator())
self.assertEqual(cu[0].name, "aa")
self.assertEqual(cu[1].name, "aa.bb")
self.assertEqual(cu[2].name, "aa.bb.cc")
self.assertEqual(cu[0].flat_rootname(), "aa")
self.assertEqual(cu[1].flat_rootname(), "aa_bb")
self.assertEqual(cu[2].flat_rootname(), "aa_bb_cc")
self.assertEqual(cu[0].source_file().read(), "# aa\n")
self.assertEqual(cu[1].source_file().read(), "# bb\n")
self.assertEqual(cu[2].source_file().read(), "") # yes, empty
def test_module_files(self):
import aa.afile, aa.bb.bfile, aa.bb.cc.cfile
cu = code_unit_factory([aa.afile, aa.bb.bfile, aa.bb.cc.cfile],
FileLocator())
self.assertEqual(cu[0].name, "aa.afile")
self.assertEqual(cu[1].name, "aa.bb.bfile")
self.assertEqual(cu[2].name, "aa.bb.cc.cfile")
self.assertEqual(cu[0].flat_rootname(), "aa_afile")
self.assertEqual(cu[1].flat_rootname(), "aa_bb_bfile")
self.assertEqual(cu[2].flat_rootname(), "aa_bb_cc_cfile")
self.assertEqual(cu[0].source_file().read(), "# afile.py\n")
self.assertEqual(cu[1].source_file().read(), "# bfile.py\n")
self.assertEqual(cu[2].source_file().read(), "# cfile.py\n")
|
