U
|
P ±d= ã @ sp U d dl mZ d dlZd dlZd dlZd dlZd dlZd dlm Z d dl
|
m Z d dl m Z mZ d dlZd dlZd dlm Z d dlmZmZmZ zd dlmZ W n ek
|
r¶ dZY nX d ZerÌd d
|
l m Z ej ej e!¡d ¡Z"ej e"d ¡Z#ej e"d ¡Z$ej e"d¡Z%ej e"d¡Z&ej e"d¡Z'ej e"d¡Z(i Z)de*d< dddddZ+dddddZ,ddd d d Z-ej.j/ej.j0ed d ej1d!dd"d#d$d%d&Z2dd'd(d)d*Z3ej.j/ej.j0ed d ej. 4d+e3e#¡d#d$d,d-Z5ej.j/ej.j0ed d ej. 4d+e3e$¡dd#d.d/d0Z6d1Z7d2Z8dddd3d4d#d5d6d7Z9d8d$d9d:Z:d8d$d;d<Z;e; Z<d8e*d=< e< =e: ¡ d>d?d@dAdBZ>ej.j/ej.j0ed d ej. 4d+e3e%¡dd#d.dCdDZ?dEZ@e AdF¡ZBddddd4d#dGdHdIZCej.j/ej.j0ed d ej. 4d+e3e#¡dd#d.dJdKZDdLdMdNdOdPdQdRdSdTdUdVdWdX ZEej.j/ej.j0ed d d#d$dYdZZFdS )[é )Ú annotationsN)Ú defaultdict)ÚIterator)ÚIOÚ TYPE_CHECKING)Ú_PRECISION_DICTÚ_EXTENDED_PRECISION_LISTÚ_C_INTP)ÚapiTF)Ú ParameterSetÚdataÚpassÚfailÚrevealÚmisczmypy.iniz .mypy_cachezdict[str, list[str]]Ú OUTPUT_MYPYÚstr)ÚkeyÚreturnc C s* t j | ¡\}}t j || dd¡d ¡S )zySplit at the first occurrence of the ``:`` character.
|
|
Windows drive-letters (*e.g.* ``C:``) are ignored herein.
|
ú:é r )ÚosÚpathÚ
|
splitdriveÚjoinÚsplit)r ÚdriveÚtail© r úUd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/typing/tests/test_typing.pyÚ _key_func. s r )Úmsgr c C s t j | ¡\}}| dd¡d S )z'Strip the filename from a mypy message.r r éÿÿÿÿ)r r r r )r! Ú_r r r r Ú_strip_filename7 s r$ z re.Match[str])Úmatchr c C s | ¡ d S )z4`re.sub` helper function for stripping module names.r )Úgroups)r% r r r Ú
|
strip_func= s r' zMypy is not installed)ÚreasonÚmodule)ZscopeZautouseÚNone)r c C sÀ t j t¡r(tt j dd¡r(t t¡ t t
|
t t fD ]} t dtdt| g¡\}}}|rht d| ¡ n |dkrt d| d| ¡ | d d
|
¡}tj| d ¡td }t d d |D ¡ q4dS )zøClears the cache and run mypy before running any of the typing tests.
|
|
The mypy results are cached in `OUTPUT_MYPY` for further use.
|
|
The cache refresh can be skipped using
|
|
NUMPY_TYPING_TEST_CLEAR_CACHE=0 pytest numpy/typing/tests
|
Z NUMPY_TYPING_TEST_CLEAR_CACHETz --config-filez --cache-dirz Unexpected mypy standard error
|
|
> r r zUnexpected mypy exit code: ú
|
|
Ú*Ú Ú
|
)r c s s" | ]\}}|r|t |fV qd S ©N)Úlist©Ú.0ÚkÚvr r r Ú <genexpr>e s zrun_mypy.<locals>.<genexpr>N)r r ÚisdirÚ CACHE_DIRÚboolÚenvironÚgetÚshutilÚrmtreeÚPASS_DIRÚ
|
REVEAL_DIRÚFAIL_DIRÚMISC_DIRr
|
ÚrunÚMYPY_INIÚpytestr ÚreplaceÚ itertoolsÚgroupbyr r r Úupdate)Ú directoryÚstdoutÚstderrZ exit_codeÚiteratorr r r Úrun_mypyB s(
|
ÿþ
|
û rL zIterator[ParameterSet])rH r c c sZ t | ¡D ]J\}}}|D ]:}t j |¡\}}|dkrt j ||¡}tj||dV qq
|
d S )N)z.pyiz.py)Úid)r Úwalkr Úsplitextr rC Úparam)rH Úrootr# ÚfilesÚfnameZ short_fnameÚextÚfullpathr r r Úget_test_casesh s rV r c C s8 t }| |kr4d}|d dd || D ¡7 }t|d S )NzUnexpected mypy output
|
|
r. c s s | ]}t |V qd S r/ )r$ )r2 r4 r r r r5 y s z test_success.<locals>.<genexpr>)r r ÚAssertionError)r Ú output_mypyr! r r r Ú test_successq s
|
rY )r r c C sD d}t | }| ¡ }W 5 Q R X tdd }t}| |ks<t|| D ]^}t| dd¡d }t d|¡}|d kr~t d| t
|
| d ¡}|| | d7 < qDt |D ]\} }
|
| d }|
|
d
|
¡s¬d |
|
krØ||krØq¬||d } d | kr$| d ¡\} } }|| ¡ }| ¡ }t| | ||| q¬t d| d|| ¡ q¬d S )NTc S s dS )Nr- r r r r r Ú<lambda> ó ztest_fail.<locals>.<lambda>r. r r z"(?P<lineno>\d+): (error|note): .+$z Unexpected error line format: Úlinenoú#z E:z# E:ú # E: z Unexpected mypy output at line r+ )ÚopenÚ readlinesr r rW r$ r Úrer% Ú
|
ValueErrorÚintÚgroupÚ enumerateÚ
|
startswithÚ partitionÚstripÚ
|
_test_failrC r )r Ú__tracebackhide__ÚfinÚlinesÚerrorsrX Ú
|
error_liner% r\ ÚiÚlineZ target_lineÚ
|
expressionr# ÚmarkerÚexpected_errorr r r Ú test_fail} sD
|
þÿþþ
|
ÿrt z9Extra error at line {}
|
|
Expression: {}
|
Extra error: {!r}
|
zTError mismatch at line {}
|
|
Expression: {}
|
Expected error: {!r}
|
Observed error: {!r}
|
z
|
None | strrc )r rq Úerrorrs r\ r c C s< |d kr t t |||¡n ||kr8t t ||||¡d S r/ )rW Ú
|
_FAIL_MSG1ÚformatÚ
|
_FAIL_MSG2)r rq ru rs r\ r r r ri · s ÿri zdict[str, str]c C sp ddddddddd d
|
d d d d } i }i }| ¡ D ]:\}}tt| }|jj|jjf}| |d| ¡||< q0|S )NÚc_ubyteÚc_ushortÚc_uintÚc_ulongÚ c_ulonglongÚc_byteÚc_shortÚc_intÚc_longÚ
|
c_longlongÚc_floatÚc_doubleÚ c_longdouble) ÚubyteÚushortÚuintcÚuintÚ ulonglongÚbyteÚshortÚintcÚint_ÚlonglongÚsingleÚdoubleÚ
|
longdoubleúctypes.)ÚitemsÚgetattrÚnpZdtypeÚkindÚitemsizeÚ
|
setdefault)ÚdctÚretÚvisitedZnp_nameZct_nameZ np_scalarr r r r Ú_construct_ctypes_dictÆ s* ó r c 0 C s dd t ¡ D } ddddddd d
|
d d d ddddddddddddddd | d d d | d d d | d d d | d! d d | d" d d | d# d d$| d d d$| d d d$| d d d$| d! d d$| d" d d$| d# d d%| d& d d%| d' d d%| d( d d%| d) d d*| d' d+| d' d d*| d( d+| d( d d*| d) d+| d) d | d" d,t d-.S ).Nc S s( i | ] \}}| d ¡d | dd¡qS )Ú.r" Únumpyz numpy._typing)r rD r1 r r r Ú
|
<dictcomp>ç s ÿ z*_construct_format_dict.<locals>.<dictcomp>z*numpy.unsignedinteger[numpy._typing._8Bit]z+numpy.unsignedinteger[numpy._typing._16Bit]z+numpy.unsignedinteger[numpy._typing._32Bit]z+numpy.unsignedinteger[numpy._typing._64Bit]z,numpy.unsignedinteger[numpy._typing._128Bit]z,numpy.unsignedinteger[numpy._typing._256Bit]z(numpy.signedinteger[numpy._typing._8Bit]z)numpy.signedinteger[numpy._typing._16Bit]z)numpy.signedinteger[numpy._typing._32Bit]z)numpy.signedinteger[numpy._typing._64Bit]z*numpy.signedinteger[numpy._typing._128Bit]z*numpy.signedinteger[numpy._typing._256Bit]z$numpy.floating[numpy._typing._16Bit]z$numpy.floating[numpy._typing._32Bit]z$numpy.floating[numpy._typing._64Bit]z$numpy.floating[numpy._typing._80Bit]z$numpy.floating[numpy._typing._96Bit]z%numpy.floating[numpy._typing._128Bit]z%numpy.floating[numpy._typing._256Bit]zAnumpy.complexfloating[numpy._typing._32Bit, numpy._typing._32Bit]zAnumpy.complexfloating[numpy._typing._64Bit, numpy._typing._64Bit]zAnumpy.complexfloating[numpy._typing._80Bit, numpy._typing._80Bit]zAnumpy.complexfloating[numpy._typing._96Bit, numpy._typing._96Bit]zCnumpy.complexfloating[numpy._typing._128Bit, numpy._typing._128Bit]zCnumpy.complexfloating[numpy._typing._256Bit, numpy._typing._256Bit]znumpy.unsignedinteger[Z _NBitByteú]Z
|
_NBitShortZ _NBitIntCZ _NBitIntPÚ_NBitIntZ _NBitLongLongznumpy.signedinteger[znumpy.floating[Z _NBitHalfZ _NBitSingleZ _NBitDoubleZ_NBitLongDoubleznumpy.complexfloating[z, r ).Zuint8Zuint16Zuint32Zuint64Úuint128Úuint256Zint8Úint16Úint32Úint64Úint128Úint256Úfloat16Úfloat32Úfloat64Úfloat80Úfloat96Úfloat128Úfloat256Z complex64Z
|
complex128Ú
|
complex160Ú
|
complex192Ú
|
complex256Ú
|
complex512r r r Zuintpr r r r r Zintpr r Zhalfr r r ZcsingleZcdoubleZ clongdoubler¢ Zc_intp)r r r )r r r r Ú_construct_format_dictæ sd ÿÁrµ Ú FORMAT_DICTzIO[str]z&tuple[npt.NDArray[np.str_], list[str]])Úfiler c C sp | ¡ dd¡}tjj| d¡ddj\}}}d |¡}tt
|
d|¡}dd |D }|j f |}|| d¡fS )
|
a Extract and parse all ``" # E: "`` comments from the passed
|
file-like object.
|
|
All format keys will be substituted for their respective value
|
from `FORMAT_DICT`, *e.g.* ``"{float64}"`` becomes
|
``"numpy.floating[numpy._typing._64Bit]"``.
|
r, r- r. r^ )Úsepz/nz \{(.*?)\}c S s" i | ]}|t |d |d¡qS )z<UNRECOGNIZED FORMAT KEY ú>)r¶ r: )r2 r3 r r r r F s ÿ z"_parse_reveals.<locals>.<dictcomp>) ÚreadrD r Úcharrg r ÚTr Úsetra Úfindallrw ) r· ÚstringÚexpression_arrayr# Zcomments_arrayÚcommentsZkey_setÚkwargsZfmt_strr r r Ú_parse_reveals3 s ÿ
|
þ rà c C s² d}t | }t|\}}W 5 Q R X t}| |ks4t|| D ]p}t|}t d|¡}|dkrjtd| t| d¡d }d|kst|| } || }
|
t
|
| |
|
| |d| q<dS )z[Validate that mypy correctly infers the return-types of
|
the expressions in `path`.
|
Tz(?P<lineno>\d+): note: .+$Nú Unexpected reveal line format: r\ r zRevealed type is) r_ rà r rW r$ ra r% rb rc rd Ú _test_reveal) r rj rk rÀ Z reveal_listrX rn r% r\ rr rq r r r Ú test_revealO s$
|
þ rÆ zWReveal mismatch at line {}
|
|
Expression: {}
|
Expected reveal: {!r}
|
Observed reveal: {!r}
|
z (\w+\.)+(\w+))r rq r Úexpected_revealr\ r c C s8 t t|¡}t t|¡}||kr4tt ||||¡dS )z2Error-reporting helper function for `test_reveal`.N)Ú_STRIP_PATTERNÚsubr' rW Ú _REVEAL_MSGrw )r rq r rÇ r\ Zstripped_revealZstripped_expected_revealr r r rÅ v s ýÿrÅ c C sv t j | ¡\}}| t j¡dd
\}}tj | d| | ¡}|dk sLt|j dk sZttj
|
|¡}|j |¡ dS )z9Validate that the code in `path` properly during runtime.éþÿÿÿNr ) r r rO r r¸ Ú importlibÚutilÚspec_from_file_locationrW ÚloaderÚmodule_from_specÚ exec_module)r Zpath_without_extensionr# ÚdirnameÚfilenameÚspecZ test_moduler r r Útest_code_runs s ÿ rÕ r£ r¤ r¨ r© r r® r¯ r° r± r² r³ r´ ) é é é é é é
|
é é é é é é c C sö t j td¡} t}| |ks tt| d}| ¡ }W 5 Q R X || D ]¬}| d¡^ }}}}t |}t
|
|} || d d¡}
|
| ¡ }|dkstt | tkrÔ|dkr¶td| qðtt | } t| |
|
| || qD|dkrDd } t| |
|
| || qDd S )
|
Nzextended_precision.pyiÚrr r r. > ru Únoteru rÄ zModule has no attribute)r r r r@ r rW r_ r` r r$ rc Úrstriprh ÚLINENO_MAPPINGr rb r¶ rÅ ri ) r rX ÚfZexpression_listÚ_msgr# Z_linenoZmsg_typr! r\ rq rr r r r Útest_extended_precision« s( rè )GÚ
|
__future__r Zimportlib.utilrÌ rE r ra r; Ú collectionsr Úcollections.abcr Útypingr r rC r r Z numpy.typingZnptZnumpy.typing.mypy_pluginr r r Zmypyr
|
Ú ImportErrorZNO_MYPYZ_pytest.mark.structuresr r r rÒ Ú__file__ZDATA_DIRr= r? r> r@ rB r7 r Ú__annotations__r r$ r' ÚmarkZslowZskipifZfixturerL rV Z parametrizerY rt rv rx ri r rµ r¶ rG rà rÆ rÊ ÚcompilerÈ rÅ rÕ rå rè r r r r Ú<module> s
|
# ) I
|
ô
|
