root/trunk/tests/test_code_parser.py

import _path_cheesecake
from cheesecake.codeparser import CodeParser
import os
datadir = os.path.abspath(os.path.join(os.path.dirname(__file__), "data"))

class TestCodeParser:
    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__"]

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

    def test_docstrings(self):
        print self.code1.docstrings
        assert self.code1.docstrings.get("module1") == 1
        for object in ["module1.Class1", "module1.Class2"]:
            assert self.code1.docstrings.get(object) == 1
        for object in ["module1.Class1.__init__", "module1.Class1.__another_method__",\
                    "module1.Class1.method1", "module1.Class1.method2", "module1.Class1.method3"]:
            assert self.code1.docstrings.get(object) == 1
        assert not self.code1.docstrings.get("module1.Class1.method4")
        for object in ["module1.func1", "module1.func2", "module1.func3", "module1.__func5__"]:
            assert self.code1.docstrings.get(object) == 1
        assert not self.code1.docstrings.get("module1.func4")