mirror of
https://github.com/kovidgoyal/calibre.git
synced 2025-07-09 03:04:10 -04:00
When finding tests also check for pyc
Needed in frozen builds
This commit is contained in:
parent
2a0bdcfcbb
commit
29bfafa676
@ -57,8 +57,14 @@ def find_tests_in_package(package, excludes=('main.py',)):
|
|||||||
loader = importlib.import_module(package).__spec__.loader
|
loader = importlib.import_module(package).__spec__.loader
|
||||||
items = list(loader.contents())
|
items = list(loader.contents())
|
||||||
suits = []
|
suits = []
|
||||||
|
excludes = set(excludes) | {x + 'c' for x in excludes}
|
||||||
|
seen = set()
|
||||||
for x in items:
|
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])
|
m = importlib.import_module(package + '.' + x.partition('.')[0])
|
||||||
suits.append(unittest.defaultTestLoader.loadTestsFromModule(m))
|
suits.append(unittest.defaultTestLoader.loadTestsFromModule(m))
|
||||||
return unittest.TestSuite(suits)
|
return unittest.TestSuite(suits)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user