a j<0 @sddlZddlZddlZddlmZddlZddlmZddl Z ddl m Z gdZ GdddZddZd d Zdd dZGdddZdS)N)_pep440)assert_)with_special_errorsassert_func_equalFuncDatac@seZdZddZdS) MissingModulecCs ||_dSN)name)selfr r >E:\soccer\ml_env\Lib\site-packages\scipy/special/_testutils.py__init__szMissingModule.__init__N)__name__ __module__ __qualname__r r r r r rsrcCsRt|tkr"tjj|jddStjjt|j t |k|j d|ddS)Nz is not installedreasonz version >= z required) typerpytestmarkskipr Zskipifrparse __version__Versionr)moduleZmin_verr r r check_versions  rcstfdd}|S)zc Enable special function errors (such as underflow, overflow, loss of precision, etc.) cs>tjdd|i|}Wdn1s00Y|S)Nraiseall)scerrstate)akwresfuncr r wrapper*s,z$with_special_errors..wrapper) functoolswraps)r%r&r r$r r%srTFc Cst|drt|}t|}|jdkr6|dddf}|jd} t|drX|} d}|}n&tj||f} tt| | jd}d}t|| tt| ||||||| || | d }| dS)Nnextr) result_columns result_funcrtolatol param_filter knownfailurenan_ok vectorizedignore_inf_signdistinguish_nan_and_inf) hasattrlistnpasarrayndimshapeZc_rangercheck)r%resultsZpointsr-r.r/r0r2dtyper1r3r4nparamsdatar+r,fdatar r r r7s*     rc @s4eZdZdZd ddZddZdd d Zd d ZdS)ra Data set for checking a special function. Parameters ---------- func : function Function to test data : numpy array columnar data to use for testing param_columns : int or tuple of ints Columns indices in which the parameters to `func` lie. Can be imaginary integers to indicate that the parameter should be cast to complex. result_columns : int or tuple of ints, optional Column indices for expected results from `func`. result_func : callable, optional Function to call to obtain results. rtol : float, optional Required relative tolerance. Default is 5*eps. atol : float, optional Required absolute tolerance. Default is 5*tiny. param_filter : function, or tuple of functions/Nones, optional Filter functions to exclude some parameter ranges. If omitted, no filtering is done. knownfailure : str, optional Known failure error message to raise when the test is run. If omitted, no exception is raised. nan_ok : bool, optional If nan is always an accepted result. vectorized : bool, optional Whether all functions passed in are vectorized. ignore_inf_sign : bool, optional Whether to ignore signs of infinities. (Doesn't matter for complex-valued functions.) distinguish_nan_and_inf : bool, optional If True, treat numbers which contain nans or infs as equal. Sets ignore_inf_sign to be True. NFTcCs||_||_| |_t|ds"|f}t||_|durdt|dsD|f}t||_|dur|d}t|n|durtd|_ntd||_||_ ||_ t|ds|f}||_ | |_ | |_ | |_| |_||_|jsd|_dS)N__len__z5Only result_func or result_columns should be providedz7Either result_func or result_columns should be providedT)r%r@datanamer5tuple param_columnsr+ ValueErrorr,r-r.r/r0r1r2r3r4)r r%r@rEr+r,r-r.r/r0rCr1r2r3r4messager r r r s:      zFuncData.__init__cCs\t|tjstt}t|}|j|j}}|durBd|j}|durTd|j }||fS)N) r7 issubdtypeZinexactr>floatZfinfor-r.epsZtiny)r r>infor-r.r r r get_tolerancess    zFuncData.get_tolerancesc) stdt}jr tjjddur.j|dur>j}n | |\}}j rt j dft j}tjj D](\}|r|t|ddfM}q|gtjD]\} t rtjddftq|r0| t|kr0ddf|| qddfqdfdd } | j} jdurtfddjD} n6d} jrt| d krt | d} | j| d } tt| t| ktt| | D]\}\}}t |jt j!sj"r2t #|}t #|}t #|}t #|}n(t $|}t $|}t %|}t %|}t |}t |}t j&d d ft '|}d|t (|<t '||}d|t (|<|t '|}d|t (|<Wdn1s0Y||||k}||k}||k}||k}||@|@|@}|j)}jr^||M}||M}|||B*8}j+sjst #|} t #|}!| |@||!@B}"||"M}||"*8}t ,|rd g}#|#d||-|#d||-|#dt *|||ft .|dD]ttfdd}$d/t0|$}%d/t0|$| }&d/t0|$| }'|$|}(|#|%d|&d|'d|(dqtdd/|#qdS)z,Check the special function against the data.Z errisinstancerNrc sjr|}nlg}ttdD]L|durF|rF|tjq$||tfddttDq$t|}t|ts|f}|S)Nrcsg|]}|qSr r ).0i)jparamsr r z?FuncData.check..eval_func_at_params..) r2r;lenappendr7nanrDr8 isinstance)r% skip_maskgot)rQr rPr eval_func_at_paramss  ,  z+FuncData.check..eval_func_at_paramscsg|]}dd|fqSrr )rNZicol)r@r r rRrSz"FuncData.check..r*)rXignorerzMax |adiff|: %gzMax |rdiff|: %gzCBad results (%d out of %d) for the following points (in output %d):csdtj|ddS)Nz%30s) precision)r7Z array2string)xrZr r fmt.szFuncData.check..fmtz z => z != z (rdiff )F )N)1operator methodcallerAssertionErrorr0rZxfailr@r>ZastyperMr/r7Zonesr:Zbool_ziprEr6 enumerate iscomplexobjintimagrUcomplexrTr%r+rDr1isnanr,rrIZcomplexfloatingr3isinfZisposinfZisneginfr absoluteisfinitesizesumr4anymaxZnonzerojoinmap))r r@r>ZdtypesZ__tracebackhide__r-r.Z param_maskfilteridxr[rYwantedrXZ output_numr`yZpinf_xZpinf_yZminf_xZminf_yZnan_xZnan_yZabs_ydiffZrdiffZtol_maskZ pinf_maskZ minf_maskZnan_maskZbad_jZ point_countZinf_xZinf_yZboth_nonfinitemsgrar!bcdr )r@rPrQr r r<s   "             0        &zFuncData.checkcCs\ttttj|jrd}nd}|jrDd|jj |t j |jSd|jj |dSdS)z%Pretty-printing, esp. for Nose outputz (complex)r]zz N) r7rsr6rvrirErCformatr%rospathbasename)r Z is_complexr r r __repr__7s  zFuncData.__repr__) NNNNNNNFTFT)NNN)rrr__doc__r rMr<rr r r r rXs( #  r) NNNNTNFFT)rr'rdZ scipy._librnumpyr7Z numpy.testingrrZ scipy.specialZspecialr__all__rrrrrr r r r s"     !