Show
Ignore:
Timestamp:
05/31/06 08:50:14 (6 years ago)
Author:
mk
Message:

Don't count empty docstrings (closes ticket #9).

Files:

Legend:

Unmodified
Added
Removed
Modified
Copied
Moved

  • branches/mk/tests/test_code_parser.py

    r11 r32  
     1import os 
     2 
    13import _path_cheesecake 
    24from cheesecake.codeparser import CodeParser 
    3 import os 
     5 
    46datadir = os.path.abspath(os.path.join(os.path.dirname(__file__), "data")) 
    57 
    6 class TestCodeParser: 
     8class TestCodeParser(object): 
    79    def setUp(self): 
    810        self.code1 = CodeParser(os.path.join(datadir, "module1.py")) 
    …… 
    1517 
    1618    def test_methods(self): 
    17         assert self.code1.methods== ["module1.Class1.__init__", 
     19        assert self.code1.methods == ["module1.Class1.__init__", 
    1820        "module1.Class1.__another_method__", 
    1921        "module1.Class1.method1", 
    …… 
    2830        "module1.func3", 
    2931        "module1.func4",  
    30         "module1.__func5__"] 
     32        "module1.__func5__", 
     33        "module1.func6"] 
    3134 
    3235    def test_count(self): 
    33         assert self.code1.object_count() == 14 
     36        assert self.code1.object_count() == 15 
    3437        assert self.code1.docstring_count() == 12 
    3538 
    3639    def test_docstrings(self): 
     40        objects_with_docstrings = [ 
     41            "module1", 
     42            "module1.Class1", 
     43            "module1.Class2", 
     44            "module1.Class1.__init__", 
     45            "module1.Class1.__another_method__", 
     46            "module1.Class1.method1", 
     47            "module1.Class1.method2", 
     48            "module1.Class1.method3", 
     49            "module1.func1", 
     50            "module1.func2", 
     51            "module1.func3", 
     52            "module1.__func5__", 
     53        ] 
     54 
     55        objects_without_docstrings = [ 
     56            "module1.Class1.method4", 
     57            "module1.func4", 
     58            "module1.func6", 
     59        ] 
     60 
    3761        print self.code1.docstrings 
    38         assert self.code1.docstrings.get("module1") == 1 
    39         for object in ["module1.Class1", "module1.Class2"]: 
    40             assert self.code1.docstrings.get(object) == 1 
    41         for object in ["module1.Class1.__init__", "module1.Class1.__another_method__",\ 
    42                     "module1.Class1.method1", "module1.Class1.method2", "module1.Class1.method3"]: 
    43             assert self.code1.docstrings.get(object) == 1 
    44         assert not self.code1.docstrings.get("module1.Class1.method4") 
    45         for object in ["module1.func1", "module1.func2", "module1.func3", "module1.__func5__"]: 
    46             assert self.code1.docstrings.get(object) == 1 
    47         assert not self.code1.docstrings.get("module1.func4") 
     62 
     63        for obj in objects_with_docstrings: 
     64            assert self.code1.docstrings.get(obj) == 1 
     65 
     66        for obj in objects_without_docstrings: 
     67            assert not self.code1.docstrings.get(obj)