mirror of
https://github.com/kennethreitz/pydantic.git
synced 2026-06-05 23:00:18 +00:00
Samuel Colvin
b7a8ef25c6
* fixing coverage by simplifying Annotated import logic, fix #2367 * avoid checking against annotated if it's None * make typing-extensions required WIP * more making typing-extensions required * fix docs and get_origin for python 3.6 * fix mypy test * fix docs * update docs, cleanup and add change * clean docs/examples/schema_annotated.py * move AnnotatedTypeNames
87 lines
3.5 KiB
Python
87 lines
3.5 KiB
Python
import importlib
|
|
import os
|
|
import re
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
try:
|
|
from mypy import api as mypy_api
|
|
except ImportError:
|
|
mypy_api = None
|
|
|
|
try:
|
|
import dotenv
|
|
except ImportError:
|
|
dotenv = None
|
|
|
|
# This ensures mypy can find the test files, no matter where tests are run from:
|
|
os.chdir(Path(__file__).parent.parent.parent)
|
|
|
|
cases = [
|
|
('mypy-plugin.ini', 'plugin_success.py', None),
|
|
('mypy-plugin.ini', 'plugin_fail.py', 'plugin-fail.txt'),
|
|
('mypy-plugin-strict.ini', 'plugin_success.py', 'plugin-success-strict.txt'),
|
|
('mypy-plugin-strict.ini', 'plugin_fail.py', 'plugin-fail-strict.txt'),
|
|
('mypy-default.ini', 'success.py', None),
|
|
('mypy-default.ini', 'fail1.py', 'fail1.txt'),
|
|
('mypy-default.ini', 'fail2.py', 'fail2.txt'),
|
|
('mypy-default.ini', 'fail3.py', 'fail3.txt'),
|
|
('mypy-default.ini', 'fail4.py', 'fail4.txt'),
|
|
('mypy-default.ini', 'plugin_success.py', 'plugin_success.txt'),
|
|
]
|
|
executable_modules = list({fname[:-3] for _, fname, out_fname in cases if out_fname is None})
|
|
|
|
|
|
@pytest.mark.skipif(not (dotenv and mypy_api), reason='dotenv or mypy are not installed')
|
|
@pytest.mark.parametrize('config_filename,python_filename,output_filename', cases)
|
|
def test_mypy_results(config_filename, python_filename, output_filename):
|
|
full_config_filename = f'tests/mypy/configs/{config_filename}'
|
|
full_filename = f'tests/mypy/modules/{python_filename}'
|
|
output_path = None if output_filename is None else Path(f'tests/mypy/outputs/{output_filename}')
|
|
|
|
# Specifying a different cache dir for each configuration dramatically speeds up subsequent execution
|
|
# It also prevents cache-invalidation-related bugs in the tests
|
|
cache_dir = f'.mypy_cache/test-{config_filename[:-4]}'
|
|
command = [full_filename, '--config-file', full_config_filename, '--cache-dir', cache_dir, '--show-error-codes']
|
|
print(f"\nExecuting: mypy {' '.join(command)}") # makes it easier to debug as necessary
|
|
actual_result = mypy_api.run(command)
|
|
actual_out, actual_err, actual_returncode = actual_result
|
|
# Need to strip filenames due to differences in formatting by OS
|
|
actual_out = '\n'.join(['.py:'.join(line.split('.py:')[1:]) for line in actual_out.split('\n') if line]).strip()
|
|
actual_out = re.sub(r'\n\s*\n', r'\n', actual_out)
|
|
|
|
if actual_out:
|
|
print('{0}\n{1:^100}\n{0}\n{2}\n{0}'.format('=' * 100, 'mypy output', actual_out))
|
|
|
|
assert actual_err == ''
|
|
expected_returncode = 0 if output_filename is None else 1
|
|
assert actual_returncode == expected_returncode
|
|
|
|
if output_path and not output_path.exists():
|
|
output_path.write_text(actual_out)
|
|
raise RuntimeError(f'wrote actual output to {output_path} since file did not exist')
|
|
|
|
expected_out = Path(output_path).read_text() if output_path else ''
|
|
assert actual_out == expected_out, actual_out
|
|
|
|
|
|
@pytest.mark.parametrize('module', executable_modules)
|
|
def test_success_cases_run(module):
|
|
"""
|
|
Ensure the "success" files can actually be executed
|
|
"""
|
|
importlib.import_module(f'tests.mypy.modules.{module}')
|
|
|
|
|
|
def test_explicit_reexports():
|
|
from pydantic import __all__ as root_all
|
|
from pydantic.main import __all__ as main
|
|
from pydantic.networks import __all__ as networks
|
|
from pydantic.tools import __all__ as tools
|
|
from pydantic.types import __all__ as types
|
|
|
|
for name, export_all in [('main', main), ('network', networks), ('tools', tools), ('types', types)]:
|
|
for export in export_all:
|
|
assert export in root_all, f'{export} is in {name}.__all__ but missing from re-export in __init__.py'
|