U md5@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_namesnameobjmsgZfailr\/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/scipy/special/tests/test_nan_inputs.py _get_ufuncss      rfunc)Zidsc 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?) rrr r Zbdtrcrrrr$)r&r'rrrtest_legacy_cast;s r*)__doc__typingrrnumpyrZ numpy.testingrrrrZ scipy.specialspecialr rstr__annotations__rrZUFUNCSZ UFUNC_NAMESrZ parametrizer(r*rrrrs