U Hv f5@sUdZddlmZmZddlZddlmZmZm Z ddl Z ddl m Z iZeeefed<iZeeefed<ddZe\ZZe jjd eed d d Zd dZdS)z3Test how the ufuncs in special handle nan inputs. )CallableDictN)assert_array_equalassert_suppress_warnings KNOWNFAILURESPOSTPROCESSINGcCsg}g}ttjD]r}tj|}t|tjs.qt|}|dkrV||||qt j j d|d}|t j ||d||q||fS)NF)runreason)Zmarks) sortedsc__dict__ isinstancenpZufuncrgetappendpytestmarkZxfailparam)ZufuncsZ ufunc_namesnameobjmsgZfailrG/tmp/pip-unpacked-wheel-96ln3f52/scipy/special/tests/test_nan_inputs.py _get_ufuncss      rfunc)idsc Cstjf|j}t\}|tdz(t}|t||}W5QRXWn tk rjYW5QRdSXW5QRX|tkrt||}d |}t t |d|ddS)N-floating point number truncated to an integerzgot {} instead of nanT)err_msg) rnanZninrfilterRuntimeWarningDeprecationWarning TypeErrorrformatrisnan)rargssupresrrrrtest_nan_inputs&s   r)c Cs@t0}|tdttjdd}tt|W5QRXdS)Nrg?) rr r!r Zbdtrcrrrr%)r'r(rrrtest_legacy_cast;s r+)__doc__typingrrZnumpyrZ numpy.testingrrrrZ scipy.specialspecialr rstr__annotations__rrZUFUNCSZ UFUNC_NAMESrZ parametrizer)r+rrrrs