mirror of
https://github.com/kevinveenbirkenbach/computer-playbook.git
synced 2025-07-03 23:52:04 +02:00
37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
# File: tests/integration/test_unittest_imports.py
|
|
|
|
import os
|
|
import unittest
|
|
|
|
class TestUnittestImports(unittest.TestCase):
|
|
TEST_ROOT = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
|
|
|
|
def test_all_test_files_import_unittest(self):
|
|
missing = []
|
|
|
|
for root, dirs, files in os.walk(self.TEST_ROOT):
|
|
for filename in files:
|
|
if not filename.endswith('.py'):
|
|
continue
|
|
# only consider test files named like "test_*.py"
|
|
if not filename.startswith('test_'):
|
|
continue
|
|
|
|
filepath = os.path.join(root, filename)
|
|
with open(filepath, encoding='utf-8') as f:
|
|
content = f.read()
|
|
|
|
# check for either import form
|
|
if 'import unittest' not in content and 'from unittest import' not in content:
|
|
rel_path = os.path.relpath(filepath, os.getcwd())
|
|
missing.append(rel_path)
|
|
|
|
if missing:
|
|
self.fail(
|
|
"The following test files do not import unittest:\n" +
|
|
"\n".join(f"- {path}" for path in missing)
|
|
)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|