Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-09 03:04:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

When finding tests also check for pyc

Browse Source 
Needed in frozen builds
This commit is contained in:
Kovid Goyal 2020-09-30 12:32:56 +05:30
parent 2a0bdcfcbb
commit 29bfafa676
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/calibre/utils/run_tests.py
View File

@ -57,8 +57,14 @@ def find_tests_in_package(package, excludes=('main.py',)):
loader = importlib.import_module(package).__spec__.loader
items = list(loader.contents())
suits = []
excludes = set(excludes) | {x + 'c' for x in excludes}
seen = set()
for x in items:
if x.endswith('.py') and x not in excludes:
if (x.endswith('.py') or x.endswith('.pyc')) and x not in excludes:
q = x.rpartition('.')[0]
if q in seen:
continue
seen.add(q)
m = importlib.import_module(package + '.' + x.partition('.')[0])
suits.append(unittest.defaultTestLoader.loadTestsFromModule(m))
return unittest.TestSuite(suits)