d.0BddlmZddlZddlmZddlZddlZddlZddlZddl m Z m Z m Z ddl Z ddlmZmZddlmZddlmZddlZddlmZdd lmZdd lmZe rdd lmZmZd Zd Z dZ!dZ"e"dddde efd(dZ#d)d*dZ$ d)d+dZ%d)d,d!Z&d)d,d"Z'd-d.d%Z(d)d/d'Z)dS)0) annotationsNwraps) TYPE_CHECKINGAnyCallable)FilePathReadPickleBuffer) get_lzma_file)import_optional_dependency)rands) ensure_clean)urlopen) DataFrameSeries) z timed outz Server Hangupz#HTTP Error 503: Service Unavailablez502: Proxy ErrorzHTTP Error 502: internal errorzHTTP Error 502zHTTP Error 503zHTTP Error 403zHTTP Error 400z$Temporary failure in name resolutionzName or service not knownzConnection refusedzcertificate verify)eonh6<cpddl}ddl}t|jjt |jjtj fS)Nr) http.client urllib.errorOSErrorclient HTTPException TimeoutErrorerrorURLErrorsockettimeout)httpurllibs O/var/www/html/t/fyr/venv311/lib/python3.11/site-packages/pandas/_testing/_io.py_get_default_network_errorsr&IsC  !   c<tfd}|S)aB allows a decorator to take optional positional and keyword arguments. Assumes that taking a single, callable, positional argument means that it is decorating a function, i.e. something like this:: @my_decorator def function(): pass Calls decorator with decorator(f, *args, **kwargs) cfd} o'tdkotd}|rd}d||S|S)Nc|gRiSN)fargs decoratorkwargss r%decz+optional_args..wrapper..decfs#9Q000000 0r'rr,)lencallable)r.r0r1 is_decoratingr-r/s`` r%wrapperzoptional_args..wrapperdsy 1 1 1 1 1 1 1# Ks4yyA~K(47:K:K  QAD3q66MJr'r)r/r6s` r% optional_argsr7Xs8 9      Nr'zhttps://www.google.comFurlstrraise_on_errorboolcheck_before_testc ddltd_tfd}|S)a Label a test as requiring network connection and, if an error is encountered, only raise if it does not find a network connection. In comparison to ``network``, this assumes an added contract to your test: you must assert that, under normal conditions, your test will ONLY fail if it does not have network connectivity. You can call this in 3 ways: as a standard decorator, with keyword arguments, or with a positional argument that is the url to check. Parameters ---------- t : callable The test requiring network connectivity. url : path The url to test via ``pandas.io.common.urlopen`` to check for connectivity. Defaults to 'https://www.google.com'. raise_on_error : bool If True, never catches errors. check_before_test : bool If True, checks connectivity before running the test case. error_classes : tuple or Exception error classes to ignore. If not in ``error_classes``, raises the error. defaults to OSError. Be careful about changing the error classes here. skip_errnos : iterable of int Any exception that has .errno or .reason.erno set to one of these values will be skipped with an appropriate message. _skip_on_messages: iterable of string any exception e for which one of the strings is a substring of str(e) will be skipped with an appropriate message. Intended to suppress errors where an errno isn't available. Notes ----- * ``raise_on_error`` supersedes ``check_before_test`` Returns ------- t : callable The decorated test ``t``, with checks for connectivity errors. Example ------- Tests decorated with @network will fail if it's possible to make a network connection to another URL (defaults to google.com):: >>> from pandas import _testing as tm >>> @tm.network ... def test_network(): ... with pd.io.common.urlopen("rabbit://bonanza.com"): ... pass >>> test_network() # doctest: +SKIP Traceback ... URLError: You can specify alternative URLs:: >>> @tm.network("https://www.yahoo.com") ... def test_something_with_yahoo(): ... raise OSError("Failure Message") >>> test_something_with_yahoo() # doctest: +SKIP Traceback (most recent call last): ... OSError: Failure Message If you set check_before_test, it will check the url first and not run the test on failure:: >>> @tm.network("failing://url.blaher", check_before_test=True) ... def test_something(): ... print("I ran!") ... raise ValueError("Failure") >>> test_something() # doctest: +SKIP Traceback (most recent call last): ... Errors not related to networking will always be raised. rNTc$r* s(t sd   |i|S#t$r}t|dd}|s&t |drt|jdd}| vrd|t |tfdDrd|t|r rd|Yd}~dSd}~wwxYw)Nzz+network..wrapper..s8II!17799 -IIIIIIr'z;Skipping test because exception message is known and error z4Skipping test due to lack of connectivity and error ) can_connectskip Exceptiongetattrhasattrr@r9any isinstance) r.r0errr?rE_skip_on_messagesr< error_classespytestr: skip_errnostr8s @r%r6znetwork..wrappers  "  ]33  KKTsTT    V1d%f%% % V V VC$//E ;WUH55 ; GT:: ## O#OOPPPHHEIIII7HIIIII  WRUWWc=11 ^  KKTsTT U U U U U U U U U% Vs8 DCD  D)rQr&networkr) rSr8r:r<rPrRrOr6rQs ``````` @r%rTrTusxMMM355 AI 1XXVVVVVVVVVVVXV> Nr'returnc|t} t|d5}|jdkr ddddS dddn #1swxYwYdS#|$rYdSwxYw)a@ Try to connect to the given url. True if succeeds, False if OSError raised Parameters ---------- url : basestring The URL to try to connect to Returns ------- connectable : bool Return True if no OSError (unable to connect) or URLError (bad url) was raised N)r"FT)r&rstatus)r8rPresponses r%rGrGs 355  S" % % % #%%        %               t uus8A A  A A AAAAA A objrpath"FilePath | ReadPickleBuffer | NoneDataFrame | Seriesc|}|dtdd}t|5}tj||tj|cdddS#1swxYwYdS)a Pickle an object and then read it again. Parameters ---------- obj : any object The object to pickle and then re-read. path : str, path object or file-like object, default None The path where the pickled object is written and then read. Returns ------- pandas object The original object that was pickled and then re-read. N__ z __.pickle)r rpd to_pickle read_pickle)r[r\_path temp_paths r%round_trip_picklergs$ E })U2YY))) e  )  S)$$$~i(())))))))))))))))))s)AA!$A! str | Nonecddl}|dj}|d}t|5}||||||}dddn #1swxYwY|S)a Write an object to file specified by a pathlib.Path and read it back Parameters ---------- writer : callable bound to pandas object IO writing function (e.g. DataFrame.to_csv ) reader : callable IO reading function (e.g. pd.read_csv ) path : str, default None The path where the object is written and then read. Returns ------- pandas object The original object that was serialized and then re-read. rNpathlib ___pathlib___)rQ importorskipPathr)writerreaderr\rQrmr[s r%round_trip_pathlibrp7s$MMM   y ) ) .D | d  !tttDzzfTT$ZZ  !!!!!!!!!!!!!!! J)A''A+.A+cddl}|dj}|d}t|5}||||||}dddn #1swxYwY|S)a Write an object to file specified by a py.path LocalPath and read it back. Parameters ---------- writer : callable bound to pandas object IO writing function (e.g. DataFrame.to_csv ) reader : callable IO reading function (e.g. pd.read_csv ) path : str, default None The path where the object is written and then read. Returns ------- pandas object The original object that was serialized and then re-read. rNzpy.path___localpath___)rQrllocalr)rnror\rQ LocalPathr[s r%round_trip_localpathrvTs$MMM##I..4I |  d  &tyyfYYt__%%&&&&&&&&&&&&&&& Jrqtestdestc<|f}d}d}|dkrtj}d}||f}d}n|dkrRtj}d}tj|}t j|} t||_|| f}d}nh|d kr tj }nU|d kr tj }nB|d krtd j}n'|d krt}nt!d||||5} t#| ||ddddS#1swxYwYdS)a Write data to a compressed file. Parameters ---------- compression : {'gzip', 'bz2', 'zip', 'xz', 'zstd'} The compression type to use. path : str The file path to write the data. data : str The data to write. dest : str, default "test" The destination file (for ZIP only) Raises ------ ValueError : An invalid compression value was passed in. wbwritezipwwritestrtar)nameaddfilegzipbz2zstd zstandardxzzUnrecognized compression type: )modeN)zipfileZipFiletarfileTarFileTarInfoioBytesIOr3sizerGzipFilerBZ2Filer openr ValueErrorrJ) compressionr\datarxr.rmethodcompress_methodfilebytesr-s r%write_to_compressedrqs&"GD D Fe!/d|   !/D))) 4  II e}   -   +   4[AAF   '//H;HHIII D ) ) )"Q6D!!""""""""""""""""""s0DDDNonecdddlm}m}||D] }||dS||dS)Nr)close get_fignums)matplotlib.pyplotrr)fignum_closers r%rrsr ~!kmm  F F6NNNN   vr')r8r9r:r;r<r;r+)rUr;)r[rr\r]rUr^)r\rh)rw)rxr9)rUr)* __future__rr functoolsrrrr!rtypingrrrrpandas._typingr r pandas.compatr pandas.compat._optionalr pandasrbpandas._testing._randomr pandas._testing.contextsrpandas.io.commonrrr_network_error_messages_network_errno_valsr&r7rTrGrgrprvrrr,r'r%rsg""""""   ('''''>>>>>>))))))111111$$$$$$ ("   :( ##-BBBBBJF:>)))))4::1"1"1"1"1"p       r'