K i LddlZddlmZmZmZddlmZmZmZm Z GddZ y)N)fixisneginfisposinf)assert_assert_array_equal assert_equal assert_raisesc*eZdZdZdZdZdZdZy) TestUfunclikectjtjtj tjdddg}tj|j t }tjgd}t|}t||t||}t||t|||jtj}tt5t|dddy#1swYyxYw)N@)TFFFFF) nparrayinfnanzerosshapeboolrrastype complex128r TypeErrorselfaouttgtress d/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/numpy/lib/tests/test_ufunclike.py test_isposinfzTestUfunclike.test_isposinf HHbffrvvgrvvsC> ?hhqww%hh@AqkS#q#S#S# HHR]] # 9 %  QK    + DD ctjtjtj tjdddg}tj|j t }tjgd}t|}t||t||}t||t|||jtj}tt5t|dddy#1swYyxYw)Nr rr)FTFFFF) rrrrrrrrrrrr rrs r test_isneginfzTestUfunclike.test_isneginfr"r#cXtjgdgdg}tj|jt}tjgdgdg}t |}t ||t ||}t ||t ||t t ddy)N)?皙?g?g?)皙gg)r'r'r'r')r)r)r)r)gQ @)rrrrfloatrrrs r test_fixzTestUfunclike.test_fix&s HH*,DE Fhhqww&hh(*>?@!fS#!SkS#S#SY"cGfddtjtjddg}|d}t|}t |tjddgt t |t|jd|d }d |_t|}t t |t|jd y) Nc(eZdZddZdfd ZdZy)5TestUfunclike.test_fix_with_subclass..MyArrayNc`tj|dj|}||_|S)NT)copy)rrviewmetadata)clsdatar5rs r __new__z=TestUfunclike.test_fix_with_subclass..MyArray.__new__4s)hht$/44S9'  r.c|t|s|j}|j|j|_|SN) isinstancer4r5)robjcontext return_scalarMyArrays r __array_wrap__zDTestUfunclike.test_fix_with_subclass..MyArray.__array_wrap__9s5!#w/((7+C<<'#'==CL r.c*t|dd|_|S)Nr5)getattrr5)rr<s r __array_finalize__zHTestUfunclike.test_fix_with_subclass..MyArray.__array_finalize__@s 'Z >  r.r:)NF)__name__ __module__ __qualname__r8r@rC)r?sr r?r13s    r.r?r(r*foo)r5)r.bar) rndarrayrrrrr;rr5)rrmfm0df0dr?s @r test_fix_with_subclassz$TestUfunclike.test_fix_with_subclass2s bjj " HHc4[ ! A & F1bhh2w/0 1g&'QZZ'i #h 3()S\\5)r.ctj}tj|}tj}t ||t t |t |d}tj |}tjd}t ||t t |t |tjd}tj ||}t||uy)Ng333333 rr )r) rrrTrue_rtyperfloat64rr)rxactualexpectedrs r test_scalarzTestUfunclike.test_scalarRs FFQ88VX&T&\4>2 ::d#VX&T&\4>2hhsms## r.N)rDrErFr!r%r-rPrXr.r r r s   #*@r.r ) numpyrrrr numpy.testingrrrr r rYr.r r\s ))RR[[r.