#ti1rUddlmZddlZddlZddlZddlmZmZmZddl m Z m Z ddl m Z mZmZddlmZmZddlZddlmZmZddlmZdd lmZdd lmZd d lmZd d lm Z m!Z!m"Z"m#Z#d dl$m%Z%d dl&m'Z'ejPdkrddl)m*Z*da+de,d<da-de,d<da.d.dZ/e d/dZ0d0dZ1d1dZ2ejfdd2dZ4ejfd d3dZ5d4d Z6ejfdd5d!Z7ejpd"e#d6d$Z9ejpd7d%Z:ejpd8d&Z;Gd'd(Z<ejpd)d9d*Z=ejpd)d9d+Z>ejpd:d,Z?ejpd;d-Z@y)<) annotationsN)Callable GeneratorIterator) ExitStackcontextmanager)isasyncgenfunctioniscoroutinefunctionismethod)Anycast)FuncFixtureInfo SubRequest)Exit) CallSpec2)Scope)get_available_backends)current_async_libraryget_async_backendreset_current_async_libraryset_current_async_library)iterate_exceptions) TestRunner) )ExceptionGroupzTestRunner | None_current_runnerzExitStack | None _runner_stackct|tr|ifSt|tr^t|dk(rPt|dtr=t|dtr*t ttttt ff|Std)Nrrz@anyio_backend must be either a string or tuple of (string, dict)) isinstancestrtuplelendictr r TypeError)backends b/Volumes/fast/ai/experiments/MLX_z-image/.venv/lib/python3.12/site-packages/anyio/pytest_plugin.pyextract_backend_and_optionsr*#sq'3{ GU #G (9 gaj# &:gaj$+Gc4S>12G< < V WWc#Ktnt|}tat %t |}tj t||xsi}tj|j|atdz a ttdza ts!tJtjdxaayy#tdza ts!tJtjdxaawwxYwwNr) rrrrrrcallbackr enter_contextcreate_test_runner_runner_leasesclose) backend_namebackend_optionsasynclibtokens r) get_runnerr7-s $\2! " *.l;E  " "#> F)/R'55  ' ' 8 aN3! , ,,    !.2 2MO ! , ,,    !.2 2MOsA>C/B: 1C/:2C,,C/c,|jdddy)N anyio_modestrictz-AnyIO plugin mode (either "strict" or "auto"))defaulthelp)addini)parsers r)pytest_addoptionr?Ks MM <r+c|jdd|jddk(rW|jjdr;|jddk(r&|j t j ddyyyy) NmarkerszManyio: mark the (coroutine function) test to be run asynchronously via anyio.r9autoasyncio asyncio_modezlAnyIO auto mode has been enabled together with pytest-asyncio auto mode. This may cause unexpected behavior.r)addinivalue_linegetini pluginmanager has_pluginissue_config_time_warningpytestPytestConfigWarning)configs r)pytest_configurerMSs W  l#v-  + +I 6 MM. )V 3((  & &<     4 7 .r+T) hookwrapperc#Kdfd }|jts tr~d|jvrp||_|j}d|jvxs|xjdz c_d|jvxs|xjdz c_ d|_||_SdS#|_||_wxYww)N anyio_backendrequestc;K|jr[trPtjt|jur&jj |j}n}t |\}}r||d< r||d<t||5}t|r|j||Ed{n|j||dddy7"#1swYyxYww)NrPrQ) instancer type__self____func____get__r*r7r run_asyncgen_fixture run_fixture) rPrQkwargs local_funcr3r4runnerfunchas_backend_arghas_request_args r)wrapperz%pytest_fixture_setup..wrapperhs   T]]#tG,<,<'==..w/?/?@JJ(CM(R% o &3F? #  'F9   o 6&!*-!66z6JJJ((V<< 7 6J7 6s0BC'!C6C7C C'CC$ C'rPrQ)rPr rQrrZr returnr )r]r r fixturenamesargnames) fixturedefrQr`original_argnamer]r^r_s @@@r)pytest_fixture_setuprhfs=4 ??D$#6t#< g22 2%JO)22 '6*:M:M'MMOM##'99#'0J4G4G'GGOG##|3# 7"& &6 # N#' &6 #sBCB2C2CC)tryfirstcx|j||rt|dr|jjn|}t |rx|j j ddk(}|jd}t|dd}|s|std|Dr%tjjd|yyyy) N hypothesisr9rBanyio pytestmarkc3:K|]}|jdk(yw)rlN)name).0markers r) z,pytest_pycollect_makeitem..sHK&v{{g-KsrP) istestfunctionhasattrrk inner_testr rLrFget_closest_markergetattranyrJmark usefixtures) collectorrpobj inner_funcanyio_auto_moderr own_markerss r)pytest_pycollect_makeitemrsT*29#|2LS^^..RU z *'..55lCvMO11':F!#|R8KHKHH ''8=I ++r+sessionc Pttt|jD]\}}t |t j s"t|js8|jdJd|jvsYg} tjtDchc]}|j}}tt#D](\}}d|vr&td|id|idt$j&i|gg}n&tid|id|idt$j&i|gg}|j(} t| j*} d| vr| j-dt/| j0| j2| | j4} t j j7|j8|j:d|d||j<| |j>|j: } |j-| +||j||d zycc}w#t$rt!}YpwxYw) NrlrP _arg2scope)paramsindicesr_idlistmarks)funcargsrr arg2scopeidlistr)re initialnames names_closurename2fixturedefs[])rpcallspeccallobj fixtureinfokeywords originalnamer) reversedlist enumerateitemsr"rJFunctionr functionrwrd dataclassesfieldsrrpr'setrrModule _fixtureinforappendrrerr from_parentparentrr}r) riitem new_itemsf cs_fields param_indexr(rfinew_names_closurenew_fixtureinfonew_items r)pytest_collection_finishrsD7==!9:;4 tV__ -#DMM2''0<t'8'88I "-8-?-? -JK-JQVV-J K)22H2J(K$ W9,( /9!0+ >$3U\\#B!(   H )!# /9!0+ >#2ELL"A 'y  H&&$()9)9$:!"*;;%,,_="1[[!#"3%'%8%8 # "??66KK --.ay:% HH /!]]!%!2!27  *O)LR(1GMM!a!e $m<L "E  "s$HH 2H HH%$H%c  d fd }|jjd}|rt|\ t|jdr`|jj j j|jk7r&t r||jj _yt|jrg|j}|jjDcic]}||| }}t 5} |j|j| dddyycc}w#t$r4}t|D] }t|t t"t$fs||d}~wwxYw#1swYyxYw)Nclt5}|j|dddy#1swYyxYwN)r7run_test)rZr\r3r4 original_funcs r)run_with_hypothesisz/pytest_pyfunc_call..run_with_hypothesiss)  o 6& OOM6 27 6 6s*3rPrkT)rZr rcNone)rgetr*rur}rkrv __qualname__r rrer7rrrr"rKeyboardInterrupt SystemExit) pyfuncitemrr(rargtestargsr\excgrpexcr3r4rs @@@r)pytest_pyfunc_callrsJ3!!%%o6G(CG(L% o :>>< 0&NN55@@M))-@-M-MM&}5;NJNN--8 z~~ .!**H6@6M6M6V6VW6VsXc]*6VHWL/:fOOJNNH=; X&1&9%cD2CZ+PQ"%61 : ;s6' D+E0D00 E-9)E(#E((E--E00E9module)scoperc|jSr)paramrbs r)rPrPs ==r+c0t|tr|S|dS)Nrr"r#ras r)anyio_backend_namer s-%Qr+c0t|triS|dSr-rras r)anyio_backend_optionsrs-% Qr+c4eZdZdZddZeddZdd dZy) FreePortFactoryaO Manages port generation based on specified socket kind, ensuring no duplicate ports are generated. This class provides functionality for generating available free ports on the system. It is initialized with a specific socket kind and can generate ports for given address families while avoiding reuse of previously generated ports. Users should not instantiate this class directly, but use the ``free_tcp_port_factory`` and ``free_udp_port_factory`` fixtures instead. For simple uses cases, ``free_tcp_port`` and ``free_udp_port`` can be used instead. c>||_tt|_yr)_kindrint _generated)selfkinds r)__init__zFreePortFactory.__init__+s c(*r+c|jS)z The type of socket connection (e.g., :data:`~socket.SOCK_STREAM` or :data:`~socket.SOCK_DGRAM`) used to bind for checking port availability )r)rs r)rzFreePortFactory.kind/szzr+NcJ||g}n@tjg}tjr|jtj d}t 5}|D]r}|j tj||j}|tjk(rdnd} |j||f|r`|jd}t||jvr&|jj||cdddSddd#t$rYwxYw#1swYxYw)z Return an unbound port for the given address family. :param family: if omitted, both IPv4 and IPv6 addresses will be tried :return: a port number Nrz::1z 127.0.0.1r) socketAF_INEThas_ipv6rAF_INET6rr/rbindOSError getsocknameradd)rfamilyfamiliesportstacksockaddrs r)__call__zFreePortFactory.__call__8s  xH'H0D&F ..v}}VTZZ/PQD$*foo$=5;D 4,/ #//1!4'4??2++D1## s7A D!D 4D7?D DDDDD")rsocket.SocketKindrcr)rcrr)rzsocket.AddressFamily | Nonercr)__name__ __module__r__doc__rpropertyrrrnr+r)rrs% %$r+r)rc4ttjSr)rr SOCK_STREAMrnr+r)free_tcp_port_factoryrZs 6-- ..r+c4ttjSr)rr SOCK_DGRAMrnr+r)free_udp_port_factoryr_s 6,, --r+c|Srrn)rs r) free_tcp_portrd ""r+c|Srrn)rs r) free_udp_portrirr+)r(objectrcztuple[str, dict[str, Any]])r3r#r4dict[str, Any]rczIterator[TestRunner])r>z pytest.Parserrcr)rLz pytest.Configrcr)rfr rQr rczGenerator[Any])r|zpytest.Module | pytest.Classrpr#r}rrcr)rzpytest.Sessionrcr)rr rcz bool | None)rQr rcr )rPr rcr#)rPr rcr)rcr)rCallable[[], int]rcr)rrrcr)A __future__rrrsyscollections.abcrrr contextlibrrinspectr r r typingr r rJ_pytest.fixturesrr_pytest.outcomesr_pytest.pythonr _pytest.scoperr_core._eventlooprrrr_core._exceptionsrabcr version_infoexceptiongrouprr__annotations__rr1r*r7r?rMhookimplrhrrrfixturerPrrrrrrrrnr+r)r s" 990EE 8!$$ 2g-%)")"& &X33(6333: &T"-#-`$>+>36>=C> > >"71t$! !Hh'='?@A    :$:$zi /!/i .!.####r+