From bf9e2566c9484b8069e8494d320b41bbdf4000a7 Mon Sep 17 00:00:00 2001 From: Karthikeyan Singaravelan Date: Sun, 1 Mar 2020 12:09:10 +0530 Subject: [PATCH] Use _read for better compatibility on using U mode. --- mccabe.py | 3 +-- test_mccabe.py | 3 +++ 2 files changed, 4 insertions(+), 2 deletions(-) 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()