U
|
P ±d ã @ s U d Z ddlmZ ddlZddlmZmZmZmZm Z m
|
Z
|
ddl Z ddl Z ddlmZ ddlmZ G dd deZejdkreejejje jZneejddZeejejjeeejejjeeejdded Z e j j d
|
e ¡ e ¡ d d dd dddZ!e j j d
|
e ¡ e ¡ d d dd dddZ"e j j d
|
e ¡ e ¡ d d dd dddZ#e j j d
|
e ¡ e ¡ d d dd dddZ$d dddZ%ej&e 'd¡fej(e )d¡fej*e )d¡fej+dgfd Z,d e-d < e j j d e, .¡ e, ¡ d G d d! d!Z/dS )"z)Test the runtime usage of `numpy.typing`.é )Ú annotationsN)Úget_type_hintsÚUnionÚ
|
NamedTupleÚget_argsÚ
|
get_originÚAnyc @ s& e Zd ZU ded< ded< ded< dS )ÚTypeTupÚtypeÚtypztuple[type, ...]Úargsz None | typeÚoriginN)Ú__name__Ú
|
__module__Ú __qualname__Ú__annotations__© r r úVd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/typing/tests/test_runtime.pyr s
|
r )é é r )Ú ArrayLikeÚ DTypeLikeÚNBitBaseÚNDArrayzname,tup)Úidsr
|
ÚNone)ÚnameÚtupÚreturnc C s& |j |j }}t|}||ks"tdS )zTest `typing.get_args`.N)r r r ÚAssertionError©r r r ÚrefÚoutr r r Ú test_get_args( s r# c C s& |j |j }}t|}||ks"tdS )zTest `typing.get_origin`.N)r r r r r r r r Útest_get_origin0 s r$ c C s@ |j }dd }|dd|_t|}|tdd}||ks<tdS )z Test `typing.get_type_hints`.c S s d S ©Nr ©Úar r r Úfunc? ó z!test_get_type_hints.<locals>.funcN©r' r ©r r r r
|
r )r r r r( r" r! r r r Útest_get_type_hints8 s r, c C sL d| |j }}dd }|dd|_t|}|tdd}||ksHtdS )zATest `typing.get_type_hints` with string-representation of types.znpt.c S s d S r% r r& r r r r( N r) z%test_get_type_hints_str.<locals>.funcNr* r+ )r r Ztyp_strr r( r" r! r r r Útest_get_type_hints_strG s r- )r c C s" t ¡ } ttj}| |ks tdS )zCTest that ``TYPES.keys()`` and ``numpy.typing.__all__`` are synced.N)ÚTYPESÚkeysÚsetÚnptÚ__all__r )r/ r! r r r Ú test_keysV s
|
r3 é é
|
)Ú_SupportsDTypeÚ_SupportsArrayÚ_SupportsArrayFuncÚ_NestedSequencez#dict[str, tuple[type[Any], object]]Ú PROTOCOLSzcls,objc @ s0 e Zd ZddddddZddddddZd S )
|
ÚTestRuntimeProtocolz type[Any]Úobjectr )ÚclsÚobjr c C s t ||stt d |r td S r% )Ú
|
isinstancer ©Úselfr= r> r r r Útest_isinstanceg s z#TestRuntimeProtocol.test_isinstancec C s< |t jkrt d¡ tt||s&tttd |r8td S )Nz<Protocols with non-method members don't support issubclass())Ú_nptr6 ÚpytestZxfailÚ
|
issubclassr
|
r r@ r r r Útest_issubclassk s
|
ÿz#TestRuntimeProtocol.test_issubclassN)r r r rB rF r r r r r; e s r; )0Ú__doc__Ú
|
__future__r ÚsysÚtypingr r r r r r rD ÚnumpyÚnpZ numpy.typingr1 Z numpy._typingÚ_typingrC r Ú version_infor Ú__args__ZndarrayZ
|
NDArrayTupr r r r. ÚmarkZ parametrizeÚitemsr/ r# r$ r, r- r3 r6 Úint64r7 Zaranger8 r9 r: r Úvaluesr; r r r r Ú<module> s@
|
ü
|
ü
|
