可以使用文档测试来测试使用get_type_hints的函数吗？

Question

我有一个使用typing.get_type_hints的函数。我想给它添加一个documentation test。但是，看起来get_type_hints无法解析文档测试中定义的类型。

下面是一个简化的示例：

import typing

def f(clazz):
    """
    >>> class MyClass:
    ...   my_field: 'MyClass'
    >>> f(MyClass)
    """
    typing.get_type_hints(clazz)

当使用python3 -m doctest test.py运行它时，它会抛出NameError: name 'MyClass' is not defined。

Answer 1

from __future__ import annotations

import typing


def f(clazz):
    """
    >>> test = 1
    >>> class MyClass:
    ...   my_field:'MyClass'
    >>> f(MyClass)
    """
    typing.get_type_hints(clazz)

在文件的开头添加from __future__ import annotations，它适用于我的python3.7