diff --git a/mccabe.py b/mccabe.py index c0cda75..3ab390c 100644 --- a/mccabe.py +++ b/mccabe.py @@ -291,8 +291,7 @@ def get_code_complexity(code, threshold=7, filename='stdin'): def get_module_complexity(module_path, threshold=7): """Returns the complexity of a module""" - with open(module_path, "rU") as mod: - code = mod.read() + code = _read(module_path) return get_code_complexity(code, threshold, filename=module_path) diff --git a/test_mccabe.py b/test_mccabe.py index f485463..b53db6a 100644 --- a/test_mccabe.py +++ b/test_mccabe.py @@ -229,6 +229,9 @@ class _options(object): mccabe.McCabeChecker.parse_options(options) self.assertEqual(16, mccabe.McCabeChecker.max_complexity) + def test_get_module_complexity(self): + self.assertEqual(0, mccabe.get_module_complexity("mccabe.py")) + if __name__ == "__main__": unittest.main()