ML i(ddlmZddlZddlZddlZddlZddlZddlmZddl m Z ddl Z ddl Z ddlmZmZmZmZddlmZmZddlmZmZGdd eZe ej4fdd Zdd Zej4d f dd ZddZddZddZ y)) annotationsN)Iterator)contextmanager) package_pathpip_lock pip_timeout test_temp_dir)DataDrivenTestCase DataSuite)assert_string_arrays_equalperform_file_operationsceZdZdgZdZddZy) PEP561Suitez pep561.test.ct|y)N) test_pep561)selftestcases Z/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/mypy/test/testpep561.pyrun_casezPEP561Suite.run_cases HNrr returnNone)__name__ __module__ __qualname__files base_pathrrrrrs OEIrrc #Ktj5}tj|dd|gt j d}|j dk7rE|jjd|jjdz}td|ztjdk(rC|tjjtjj|d d fnB|tjjtjj|d d fd d d y #1swYy xYww) z~Context manager that creates a virtualenv in a temporary directory Returns the path to the created Python executable -mvenvT)cwdcapture_outputrutf-8zFailed to create venv. win32ScriptspythonbinN)tempfileTemporaryDirectory subprocessrunosgetcwd returncodestdoutdecodestderr Exceptionsysplatformpathabspathjoin)python_executablevenv_dirprocerrs r virtualenvr?s  $ $ & U(~~ fh 7RYY[Y]  ??a ++$$W- 0B0B70KKC6<= = <<7 "BGGOOBGGLL9h,WXX XBGGOOBGGLL5(,STT T U U UsEDD;2 E;EEcXtjdk\s7dtjcxkrdksndtjcxkrdkryny|dddd g} tjtt 5t j|d tj }dddjdk7r@t|jjd|jjdzy#1swYYxYw#tj$r}td t|d}~wwxYw)zAInstall pip>=21.3.1. Required for editable installs with PEP 660.) )rA rA)rA rB)rArCNr"pipinstallz pip>=21.3.1timeoutT)r%envFailed to acquire rr&)r6 version_infofilelockFileLockrrr-r.r/environTimeoutr5r1r2r3r4)r; install_cmdr=r>s r upgrade_piprQ1s  G# )) 3G 3 )) 3G 3  4 $dE9mLKB   x = T>>+d SD T ! **73dkk6H6H6QQRR  T T   B,XJ78cABs0C>3'C2C>2C;7C>>D)D$$D)Fctjjt|}t j 5}|dddg}|r|j d|j dd|i}|jtj tjtt5tj||d| }d d d d d d j"d k7r@t!|j$j'd |j(j'd zy #1swYaxYw#tj$r}t!d t|d }~wwxYw#1swYxYw)z.Install a package from test-data/packages/pkg/r"rErFz-er PIP_BUILDrGT)r$r%rINrJrr&)r/r8r:rr+r,appendupdaterNrLrMrrr-r.rOr5r1r2r3r4) pkgr;editable working_dirdirrPrIr=r>s rinstall_packagerZFs:'',,|S1K  $ $ &F#($yA    t $3C  2:: F""8[A b!~~k{SW]`a bF  ! **73dkk6H6H6QQRR  b b F0 ;<# E FFFsBAED-'D!D-!D* &D--EEEEE$c |jJdtj}|Jdt|jd\}}t |jd}d}|D]}|dk(rd} t d ||sJd t|5}|\}} |r t| |D]} t| | |t|} d | vxsd | v } | r]|jd z} t| dd5}|jD]}|j|d ddd| j| | jddg| tjk7r| jd| |j!}|ggk7rgg|z}t#|D]\}}t%|g}t&j(j+| \}}}||zj-D]}|j/t0t2j4zrB|j|t7t0t2j4zdj9dk|j;dd^}}|r |j=t2j4d}|jdj?|g|j9d|dk(rdnd|dz}|dk(r |j@n|jBjE|dzg}tG||d|jHd|jJd|| rt3jL dddy#1swY(xYw#1swYyxYw)z;Test running mypy on files that depend on PEP 561 packages.Nztest was not properly set upzShould be impossiblerFrWTzUnknown pip argument: z(No packages to install for PEP 561 test?z-pz --packagez.pywr&)encoding z--no-error-summaryz--hide-error-codesz--python-executable=z :)maxsplit/z on iteration zInvalid output (z, line ))'old_cwdr6 executable parse_pkgsinputparse_mypy_args ValueErrorr?rQrZlistnameopenwriterTextend find_steps enumerater mypyapir. splitlines startswithr r/seplenrstripsplitreplacer:outputoutput2getr filelineremove)rr)pkgspip_args mypy_argsrWargr#r<r;rVcmd_line has_programprogramfsstepsi operationsr{outr>r1rr8rest iter_countexpecteds rrr_sP    'G)GG ' ^^F  555 q 12ND(q 12IH= * H5cU;< < = ;;;4 F 5t&*##  ) * >C C!2H = > ?8+F{h/FG mme+GgsW5 &!&AGGqcH%& & OOG $-/CDE  . OO23D2EF G##% RD=D5LE&u- MAz #J /F#'88<<#9 Cjs..0 J??=266#9:MM$s=266+A'B'D"E"L"LV"TU#'**S1*"=KD4#||BFFC8MM#((D=4="9"@"@"HI J !Av^AE7+CJ*+q&xh6F6F6J6J1q5RT6UH &"8==/q U + 6  IIg k55 & &55s&AM.&L;HM;M MMc|jdsggfS|ddjd^}}|jdDcgc]}|jc}|Dcgc]}|jc}fScc}wcc}w)Nz# pkgs:;,)rurystrip)commentpkgs_strargsrVrs rrgrgsp   i (Bx!!"+++C04(0s(;<[ssyy{>[\\<>[s A7A<cttjd|}|sgS|jdjS)Nz# flags: (.*)$r\)rematchgroupry)rms rriris2 !4(A  771:   r)r;strrzIterator[tuple[str, str]])r;rrr)rVrr;rrWboolrrr)rrrztuple[list[str], list[str]])rrrz list[str])! __future__rr/rr-r6r+collections.abcr contextlibrrLmypy.apirrmypy.test.configrrrr mypy.test.datar r mypy.test.helpersr r rrfr?rQrZrrgrir rrrs"  $%OO8Q )(+UU$S,(+~~S S!$SAES S2DN]r