a :jg @sFddlZddlmZmZmZddlmZmZmZm Z GdddZ dS)N)fixisposinfisneginf)assert_ assert_equalassert_array_equal assert_raisesc@s4eZdZddZddZddZddZd d Zd S) TestUfunclikecCsttjtj tjdddg}t|jt}tgd}t|}t||t||}t||t||| tj }t t t|Wdn1s0YdS)N@)TFFFFF) nparrayinfnanzerosshapeboolrrastype complex128r TypeErrorselfaoutZtgtresrH/usr/local/lib/python3.9/site-packages/numpy/lib/tests/test_ufunclike.py test_isposinf s      zTestUfunclike.test_isposinfcCsttjtj tjdddg}t|jt}tgd}t|}t||t||}t||t||| tj }t t t|Wdn1s0YdS)Nr r r )FTFFFF) r rrrrrrrrrrrrrrrr test_isneginfs      zTestUfunclike.test_isneginfcCs|tgdgdg}t|jt}tgdgdg}t|}t||t||}t||t||ttdddS)N)?皙?g?g?)皙gg)r r r r )r"r"r"r"gQ @)r rrrfloatrrrrrrtest_fix)s    zTestUfunclike.test_fixcsGfdddtjtddg}|dd}t|}t|tddgtt|t|jd|d }d |_t|}tt|t|jd dS) Ncs,eZdZd ddZd fdd ZddZdS) z5TestUfunclike.test_fix_with_subclass..MyArrayNcSstj|dd|}||_|S)NT)copy)r rviewmetadata)clsdatar)rrrr__new__7sz=TestUfunclike.test_fix_with_subclass..MyArray.__new__Fcs*t|s|}|jdur&|j|_|S)N) isinstancer(r))robjcontextZ return_scalarMyArrayrr__array_wrap__<s    zDTestUfunclike.test_fix_with_subclass..MyArray.__array_wrap__cSst|dd|_|S)Nr))getattrr))rr.rrr__array_finalize__CszHTestUfunclike.test_fix_with_subclass..MyArray.__array_finalize__)N)NF)__name__ __module__ __qualname__r,r2r4rr0rrr16s r1r!r#Zfoo)r))r.bar) r Zndarrayrrrrr-rr))rrmfZm0dZf0drr0rtest_fix_with_subclass5s  z$TestUfunclike.test_fix_with_subclasscCstj}t|}tj}t||tt|t|d}t|}td}t||tt|t|td}tj||d}t ||udS)Ng333333 r r )r) r rrZTrue_rtyperZfloat64rr)rxactualexpectedrrrr test_scalarUs      zTestUfunclike.test_scalarN)r5r6r7rrr&r=rBrrrrr s   r ) numpyr rrrZ numpy.testingrrrrr rrrrs