root/branches/mk/tests/test_code_parser.py

import os

import _path_cheesecake
from cheesecake.codeparser import CodeParser

datadir = os.path.abspath(os.path.join(os.path.dirname(__file__), "data"))

class TestCodeParser(object):
    def setUp(self):
        self.code1 = CodeParser(os.path.join(datadir, "module1.py"))

    def test_modules(self):
        assert self.code1.modules == ["module1"]

    def test_classes(self):
        assert self.code1.classes == ["module1.Class1", "module1.Class2"]

    def test_methods(self):
        assert self.code1.methods == ["module1.Class1.__init__",
        "module1.Class1.__another_method__",
        "module1.Class1.method1",
        "module1.Class1.method2", 
        "module1.Class1.method3",
        "module1.Class1.method4"]

    def test_functions(self):
        assert self.code1.functions == [
        "module1.func1", 
        "module1.func2", 
        "module1.func3",
        "module1.func4", 
        "module1.__func5__",
        "module1.func6"]

    def test_count(self):
        assert self.code1.object_count() == 15
        assert self.code1.docstring_count() == 12

    def test_docstrings(self):
        objects_with_docstrings = [
            "module1",
            "module1.Class1",
            "module1.Class2",
            "module1.Class1.__init__",
            "module1.Class1.__another_method__",
            "module1.Class1.method1",
            "module1.Class1.method2",
            "module1.Class1.method3",
            "module1.func1",
            "module1.func2",
            "module1.func3",
            "module1.__func5__",
        ]

        objects_without_docstrings = [
            "module1.Class1.method4",
            "module1.func4",
            "module1.func6",
        ]

        print self.code1.docstrings

        for obj in objects_with_docstrings:
            assert self.code1.docstrings.get(obj) == 1

        for obj in objects_without_docstrings:
            assert not self.code1.docstrings.get(obj)