1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
# (C) Copyright 2020-2023 Enthought, Inc., Austin, TX
# All rights reserved.
#
# This software is provided without warranty under the terms of the BSD
# license included in LICENSE.txt and may be redistributed only under
# the conditions described in the aforementioned license. The license
# is also available online at http://www.enthought.com/licenses/BSD.txt
#
# Thanks for using Enthought open source!
from pathlib import Path
from unittest import TestCase
from traits.testing.optional_dependencies import (
pkg_resources,
requires_mypy,
requires_numpy_typing,
requires_pkg_resources,
)
from traits_stubs_tests.util import MypyAssertions
@requires_pkg_resources
@requires_mypy
class TestAnnotations(TestCase, MypyAssertions):
def test_all(self, filename_suffix=""):
""" Run mypy for all files contained in traits_stubs_tests/examples
directory.
Lines with expected errors are marked inside these files.
Any mismatch will raise an assertion error.
Parameters
----------
filename_suffix: str
Optional filename suffix filter.
"""
examples_dir = Path(pkg_resources.resource_filename(
'traits_stubs_tests', 'examples'))
for file_path in examples_dir.glob("*{}.py".format(filename_suffix)):
with self.subTest(file_path=file_path):
self.assertRaisesMypyError(file_path)
@requires_numpy_typing
def test_numpy_examples(self):
""" Run mypy for files contained in traits_stubs_tests/numpy_examples
directory.
Lines with expected errors are marked inside these files.
Any mismatch will raise an assertion error.
Parameters
----------
filename_suffix: str
Optional filename suffix filter.
"""
examples_dir = Path(pkg_resources.resource_filename(
'traits_stubs_tests', 'numpy_examples'))
for file_path in examples_dir.glob("*.py"):
with self.subTest(file_path=file_path):
self.assertRaisesMypyError(file_path)
|
