U md @sTddlZddlmZddlmmZddl m Z m Z m Z m Z mZGdddZdS)N)assert_ assert_equalassert_array_equal assert_warns assert_raisesc@s<eZdZddZddZddZddZd d Zd d Zd S) TestUfunclikec Csttjtj tjdddg}t|jt}tddddddg}t|}t ||t||}t ||t ||| t j }t tt|W5QRXdS)N@TF)nxarrayinfnanzerosshapebooluflisposinfrastypenpcomplex_r TypeErrorselfaoutZtgtresrW/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/numpy/lib/tests/test_ufunclike.py test_isposinf s       zTestUfunclike.test_isposinfc Csttjtj tjdddg}t|jt}tddddddg}t|}t ||t||}t ||t ||| t j }t tt|W5QRXdS)Nrr r FT)r r r rrrrrisneginfrrrrrrrrrr test_isneginfs       zTestUfunclike.test_isneginfcCstddddgddddgg}t|jt}tddddgddddgg}t|}t||t||}t||t||ttd d dS) Ng?皙?g?g?g皙gggQ @)r r rrfloatrfixrrrrrtest_fix)s     zTestUfunclike.test_fixcsGfdddtjtddg}|dd}t|}t|tddgtt|t|j d|d }d |_ t|}tt|t|j d dS) Ncs,eZdZdddZd fdd ZddZdS) z5TestUfunclike.test_fix_with_subclass..MyArrayNcSstj|dd|}||_|S)NT)copy)r r viewmetadata)clsdatar*rrrr__new__7sz=TestUfunclike.test_fix_with_subclass..MyArray.__new__cst|r|j|_|S)N) isinstancer*)robjcontextMyArrayrr__array_wrap__<s zDTestUfunclike.test_fix_with_subclass..MyArray.__array_wrap__cSst|dd|_|S)Nr*)getattrr*)rr/rrr__array_finalize__AszHTestUfunclike.test_fix_with_subclass..MyArray.__array_finalize__)N)N)__name__ __module__ __qualname__r-r3r5rr1rrr26s r2r"r#Zfoo)r*)r.bar) r Zndarrayr rr&rrr.rr*)rrmfZm0dZf0drr1rtest_fix_with_subclass5s    z$TestUfunclike.test_fix_with_subclasscCsXtttjddgtddtttjddgtddtttjddgtdddS)Nr9)y)rDeprecationWarningrr&r emptyrr )rrrrtest_deprecatedSszTestUfunclike.test_deprecatedcCstj}t|}tj}t||tt|t|d}t|}td}t||tt|t|td}tj||d}t ||kdS)Ng333333 r r)r) rr rZTrue_rtyper&Zfloat64r r)rxactualexpectedrrrr test_scalarYs      zTestUfunclike.test_scalarN) r6r7r8rr!r'r>rCrHrrrrr s  r)numpyrZ numpy.corecorer Znumpy.lib.ufunclikelibZ ufunclikerZ numpy.testingrrrrrrrrrrs