def test_raw_visitor_functions(code, expected):
code = test_raw.dedent(code)
raw_visitor = RawVisitor.from_code(code)
# only one function in these tests
raw_result = raw_visitor.functions[0]
# exclude the details about function name, lineno, etc. for now
formated_result = Module(*raw_result[7:])
assert formated_result == Module(*expected), f"\
\n input code: {code}\
\n result: {formated_result} \
\n expected: {Module(*expected)}"
expected_loc = (formated_result.blank + formated_result.sloc +
formated_result.single_comments + formated_result.multi)
assert formated_result.loc == expected_loc
def test_raw_to_dict():
assert tools.raw_to_dict(Module(103, 123, 98, 8, 19, 5, 3)) == {
'loc': 103,
'lloc': 123,
'sloc': 98,
'comments': 8,
'multi': 19,
'blank': 5,
'single_comments': 3,
}
def test_raw_to_dict(self):
self.assertEqual(
tools.raw_to_dict(Module(103, 123, 98, 8, 19, 5)), {
'loc': 103,
'lloc': 123,
'sloc': 98,
'comments': 8,
'multi': 19,
'blank': 5
})