test_annotations.py 文件源码

def test_short_comment_types(create_project, run):
    """Test type hinting with short-form comments."""
    with create_project('''
        from typing import Any, AnyStr

        def types(str1, num):
            # type: (AnyStr, int) -> Any
            """usage: say types <str1> <num>"""
            print(type(str1))
            print(type(num))
    '''):
        type_reprs = run('say types world 4', stderr=True).strip().split('\n')
        assert type_reprs == [
            repr(str),
            repr(int)
        ]