o 6a`@sddlZddlmZddlmZmZmZmZm Z ddiZ ddiZ GdddZ Gd d d Z Gd d d ZGd ddZGdddZGdddeZGdddeZGdddZGdddZGdddZddZdS)N _vec_string)assert_ assert_equalassert_array_equal assert_raisesassert_raises_regexunicodeTFc@DeZdZddZddZddZddZd d Zd d Zd dZ dS) TestBasiccCsNtjddgddggdd}tj|}t|jjdt|dd gd d ggdS) Nabclong 0123456789Odtype abc2slong 0123456789)nparraycharrritemsizerselfABrD/usr/lib/python3/dist-packages/numpy/core/tests/test_defchararray.pytest_from_object_array s    z TestBasic.test_from_object_arraycCsxtjddgddggdd}tttjj|ftjj|fit}t|jjdtdd jjt |ddgd dggdS) Nr Sigma ΣrrrrraUlong) rrr ValueErrorrkw_unicode_truerrrrrrrr test_from_object_array_unicodes   z(TestBasic.test_from_object_array_unicodecCstddgddgg}t|jjtjtj|}t||t|j|jt|j|jd|d<t |d|dktj |}t||t|j|jd|d<t |d|dkt |d|dkdS)Nrsfooslong rchangedrrz changed again) rrrrtypestring_rrshaperasarray)rrrCrrr test_from_string_arrays     z TestBasic.test_from_string_arraycstddgddggtjjtjtj}t|t|jjt|jjtjjfit }t|t|jjt|jjfdd}t t |dS)Nr r"rrcstjjfitdSN)rrrkw_unicode_falserrrr fail<sz/TestBasic.test_from_unicode_array..fail) rrrrr+unicode_rrr-r'rUnicodeEncodeError)rrr4rr3r test_from_unicode_array/s     z!TestBasic.test_from_unicode_arraycCs8tjdg}tjdg}tt||jjtjdS)Nr Σ)rrrr issubclassrr+r5rrrr test_unicode_upconvertAsz TestBasic.test_unicode_upconvertcCsDtjd}tt|dtt|ddtt|jjtj dS)Nrr) rrrrlenrr9rr+r,rrrrr test_from_stringFs zTestBasic.test_from_stringcCsPtjd}tt|dtt|ddt|jdtt|jj tj dS)Nr8r;r) rrrrr=rrr9rr+r5r>rrr test_from_unicodeLs  zTestBasic.test_from_unicodeN) __name__ __module__ __qualname__r!r(r0r7r:r?rArrrr r s  r c@r ) TestVecStringcCdd}tt|dS)NcStdtjddS)Nr#bogusrrr,rrrr r4Vz4TestVecString.test_non_existent_method..fail)rAttributeErrorrr4rrr test_non_existent_methodTz&TestVecString.test_non_existent_methodcCrF)NcSrG)Nr;striprIrrrr r4]rJz1TestVecString.test_non_string_array..failr TypeErrorrLrrr test_non_string_array[rNz#TestVecString.test_non_string_arraycCrF)NcStdgtjdddS)Nr#rOr;rIrrrr r4dz3TestVecString.test_invalid_args_tuple..failrPrLrrr test_invalid_args_tuplebrNz%TestVecString.test_invalid_args_tuplecCrF)NcSstdgdddS)Nr#BOGUSrOrrrrr r4krJz3TestVecString.test_invalid_type_descr..failrPrLrrr test_invalid_type_descrirNz%TestVecString.test_invalid_type_descrcCrF)NcSrS)Nr#rOr;rIrrrr r4rrTz6TestVecString.test_invalid_function_args..failrPrLrrr test_invalid_function_argsprNz(TestVecString.test_invalid_function_argscCrF)NcSstdgtjddS)Nr#rOrrint_rrrr r4ysz4TestVecString.test_invalid_result_type..failrPrLrrr test_invalid_result_typewrNz&TestVecString.test_invalid_result_typecCrF)NcSs tddggtjdgdfdS)Nr deffind)r#djrZrrrr r4s z0TestVecString.test_broadcast_error..fail)rr&rLrrr test_broadcast_error~rNz"TestVecString.test_broadcast_errorN) rBrCrDrMrRrUrWrYr\rarrrr rESs rEc@eZdZddZddZdS)TestWhitespacecCDtddgddggtj|_tddgddggtj|_dS) Nabc 123 z789 zxyz r 123789xyzrrview chararrayrrrrrr setup  zTestWhitespace.setupcCstt|j|jktt|j|jktt|j|jktt|j|jk tt|j|jk tt|j|jk dSr1)rrallrranyrmrrr test1s zTestWhitespace.test1N)rBrCrDrnrrrrrr rcs rcc@rb)TestCharcCstjdddtj|_dS)Nabc1cr)rrrkrlrrmrrr rnszTestChar.setupcCs.t|jjdt|jddddS)Nr@r sAB)rrr-uppertobytesrmrrr test_its zTestChar.test_itN)rBrCrDrnryrrrr rss rsc@LeZdZddZddZddZddZd d Zd d Zd dZ ddZ dS)TestComparisonscCrd) Nr rgrhriefgrf051tuvrjrmrrr rnrozTestComparisons.setupcCs"t|j|jkddgddggdSNTFrrrrmrrr test_not_equal"zTestComparisons.test_not_equalcCs"t|j|jkddgddggdSNFTrrmrrr test_equalrzTestComparisons.test_equalcCs"t|j|jkddgddggdSrrrmrrr test_greater_equalrz"TestComparisons.test_greater_equalcCs"t|j|jkddgddggdSrrrmrrr test_less_equalrzTestComparisons.test_less_equalcCs"t|j|jkddgddggdSrrrmrrr test_greaterrzTestComparisons.test_greatercCs"t|j|jkddgddggdSrrrmrrr test_lessrzTestComparisons.test_lesscCsDtj|j|j}tjdd}tt|tjtt|tjdSNr#)rrequalrrr isinstancendarray)rout1out2rrr test_typeszTestComparisons.test_typeN) rBrCrDrnrrrrrrrrrrr r{s r{c@eZdZdZddZdS)TestComparisonsMixed1 Ticket #1276cC2t|tddgddggtjtj|_dS)Nr|rfr}r~)r{rnrrr5rkrlrrmrrr rn  zTestComparisonsMixed1.setupNrBrCrD__doc__rnrrrr r rc@r)TestComparisonsMixed2rcCr)Nr rgrhri)r{rnrrr5rkrlrrmrrr rnrzTestComparisonsMixed2.setupNrrrrr rrrc@seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!S)"TestInformationcCsPtddgddgddggtj|_tddgddgddggtj|_dS)N abc 12345 MixedCase 123 345 UPPER Σ rjrmrrr rns  zTestInformation.setupcCsntttj|jjjtjt tj|jddgddgddggt tj|j ddgddgddggdS)Nr r<) rr9rrstr_lenrrr+integerrrrmrrr test_lens&*zTestInformation.test_lencCstt|jdjjtjt|jdddgddgddggt|jdddgddgddggt|jdddddgddgddggt|j dddgddgddggt|j dddgddgddggdS)Nrr#r;rrgr ) rr9rcountrr+rrrrrmrrr test_counts $$($(zTestInformation.test_countcsttjdjjtjtjdddgddgddggtjdddddgddgddggfdd}t t |dS) Nr r;r3r<cjdddSNrfdjk)rendswithrrmrr r4rJz+TestInformation.test_endswith..fail) rr9rrrr+rbool_rrrQrLrrmr test_endswith $( zTestInformation.test_endswithcCstt|jdjjtjt|jdddgddgddggt|jdddgddgddggt|jdddddgddgddggt|jdd gddgddgddggdS Nr#r;rr r1P) rr9rr^rr+rrrrmrrr test_find $$(,zTestInformation.test_findcNfdd}tt|ttjdddktttjddjjtj dS)NcjddSr)rindexrrmrr r4z(TestInformation.test_index..failabcbabr;) rr&rrrrr9rr+rrLrrmr test_index  "zTestInformation.test_indexcCs@tt|jjjtjt|jddgddgddggdSr) rr9risalnumrr+rrrrmrrr test_isalnum&zTestInformation.test_isalnumcCs@tt|jjjtjt|jddgddgddggdSr) rr9risalpharr+rrrrmrrr test_isalpharzTestInformation.test_isalphacCs@tt|jjjtjt|jddgddgddggdSr) rr9risdigitrr+rrrrmrrr test_isdigitrzTestInformation.test_isdigitcCs@tt|jjjtjt|jddgddgddggdSr) rr9rislowerrr+rrrrmrrr test_islower rzTestInformation.test_islowercC@tt|jjjtjt|jddgddgddggdSNF) rr9risspacerr+rrrrmrrr test_isspacerzTestInformation.test_isspacecCrr) rr9ristitlerr+rrrrmrrr test_istitlerzTestInformation.test_istitlecCs@tt|jjjtjt|jddgddgddggdSr) rr9risupperrr+rrrrmrrr test_isupperrzTestInformation.test_isuppercCstt|jdjjtjt|jdddgddgddggt|jdddgddgddggt|jdddddgddgddggt|jdd gddgddgddggdSr) rr9rrfindrr+rrrrmrrr test_rfindrzTestInformation.test_rfindcr)Ncrr)rrindexrrmrr r4$rz)TestInformation.test_rindex..failrrr<) rr&rrrrr9rr+rrLrrmr test_rindex"rzTestInformation.test_rindexcsttjdjjtjtjdddgddgddggtjdddddgddgddggfdd}t t |dS) Nrrr;rrr<crr)r startswithrrmrr r40rJz-TestInformation.test_startswith..fail) rr9rrrr+rrrrrQrLrrmr test_startswith+rzTestInformation.test_startswithN)rBrCrDrnrrrrrrrrrrrrrrrrrrr rs"    rc@seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1S)2 TestMethodscCsTtjddgddgddggddtj|_td dgddgddggtj|_dS) NrrrrrrSrrrjrmrrr rn7s  zTestMethods.setupcCddgddgddgg}tt|jjjtjt|j|ddgd d gd d gg}tt|j jjtj t|j |dS) N abc 12345 Mixedcase 123 345 Upper σ rr MixedcaserUpper) rr9r capitalizerr+rr,rrr5rtgtrrr test_capitalize@zTestMethods.test_capitalizecCstt|jdjjtj|jddg}ttj |ddgddgddgg|jdd}tt | dtt | dtj dddgddgg}dd gd d gg}tt|jjtjt||dS) Nrr#FOOs FOO s FOO s FOO s FOO )rr9rcenterrr+rr,rrrrprrrr/rrrr test_centerMs$zTestMethods.test_centercCs(tjdg}t|dddkdS)Ns\u03a3zunicode-escaperr8)rrrrdecoder>rrr test_decode\szTestMethods.test_decodecCs.|jd}t|ddtddkdS)Nunicode_escaperz \u03a3 latin1)rencoderstr)rrrrr test_encode`s "zTestMethods.test_encodecCs|j}t|ddkdS)N)r rs123 345 )r expandtabsr)rTrrr test_expandtabsds zTestMethods.test_expandtabscCsj|jd}tjddg|}tt|jjtj t ddgddgdd gg}t tjddg||dS) Nascii,#z ,a,b,c, rz 1,2,3,4,5zM#i#x#e#d#C#a#s#ez1,2,3, , , ,3,4,5, ,, z U#P#P#E#R) rrrrjoinrr9rr+r5rr)rA0rrrrr test_joinhs  zTestMethods.test_joincCstt|jdjjtj|jddg}ttj |ddgddgddgg|jdd}t| dddgddgddggtt | dtj dddgdd gg}d d gd d gg}tt|jjtjt||dS)NrrrrFTrrrs FOO sFOO sFOO sFOO )rr9rljustrr+rr,rrrrrprrrrr test_ljustts$ zTestMethods.test_ljustcCr) Nrrrs mixedcaserupperrrr mixedcaserrw) rr9rlowerrr+rr,rrr5rrrr test_lowerrzTestMethods.test_lowercCsddgddgddgg}tt|jjjtjt|j|ddgdd gd dgg}t|jd d g|d dgddgddgg}tt|j jjtj t|j |dS)Nsabc rr MixedCaserUPPER abcs2345ixedCase 23 345 1MuΣ rrrrr) rr9rlstriprr+rr,rrr5rrrr test_lstrip"zTestMethods.test_lstripcCH|jddg}ddgddgddgg}tt|jjtjt||dS) N3r )rrrrrr12r45rr r )rrs 345 )rrr) r partitionrr9rr+rr,rrrrrrr test_partitionzTestMethods.test_partitioncCsN|jddgddg}ddgddgd d gg}tt|jjtjt||dS) Nras ##########@rrs12##########45s MixedC@ses12########## ##########45 r) rreplacerr9rr+rr,r)rRrrrr test_replaces zTestMethods.test_replacecCstt|jdjjtj|jddg}ttj |ddgddgddgg|jdd}tt | dt| dddgddgddggtj dddgdd gg}d d gd d gg}tt|jjtjt||dS)NrrrrFTrrrs FOOs FOOs FOOs FOO)rr9rrjustrr+rr,rrrrprrrrrr test_rjusts$ zTestMethods.test_rjustcCr) Nrr )rrrrrr)s123 r45 )rrr) r rpartitionrr9rr+rr,rrrrr test_rpartitionrzTestMethods.test_rpartitioncCX|jd}dgdggddgdgggddggg}tt|jjtjt| |dS Nrrrrrr)rs r#r) rrsplitrr9rr+robject_rtolistrrrrrr test_rsplits    zTestMethods.test_rsplitcCstt|jjjtjddgddgddgg}t|j|ddgddgd d gg}t|jd d g|d dgddgddgg}tt|j jjtj t|j |dS)Nr rrr 123 345rrs1234s 123 345 UPP5sERu Σrrr 123 345r) rr9rrstriprr+rr,rrr5rrrr test_rstrips"zTestMethods.test_rstripcCsddgddgddgg}tt|jjjtjt|j|ddgdd gd d gg}t|jd d g|ddgddgddgg}tt|j jjtj t|j |dS)Nrrrrr-rrs234sixedCasr r.s15sEReMr8rrrr0r) rr9rrOrr+rr,rrr5rrrr test_striprzTestMethods.test_stripcCr&r') rsplitrr9rr+rr)rr*r+rrr test_splits    zTestMethods.test_splitcCsLtjdg}tt|jjtjt|j dktt |ddkdS)Nz abc fds werrXrr<) rrr splitlinesrr9rr+r)r-r=r>rrr test_splitlinesszTestMethods.test_splitlinescCr) N ABC rrs mIXEDcASErrrrr mIXEDcASErrw) rr9rswapcaserr+rr,rrr5rrrr test_swapcase rzTestMethods.test_swapcasecCr) Ns Abc rrrrrrrrrrr) rr9rtitlerr+rr,rrr5rrrr test_titlerzTestMethods.test_titlecCr) Nr8rrs MIXEDCASErrrrr MIXEDCASErr) rr9rrwrr+rr,rrr5rrrr test_upper%rzTestMethods.test_uppercVfdd}tt|ttjjjtj t jddgddgddggdS)NcjdSr1)r isnumericrrmrr r44z(TestMethods.test_isnumeric..failFT) rrQrr9rrBrr+rrrrLrrmr test_isnumeric2    zTestMethods.test_isnumericcr@)NcrAr1)r isdecimalrrmrr r4>rCz(TestMethods.test_isdecimal..failFT) rrQrr9rrFrr+rrrrLrrmr test_isdecimal<rEzTestMethods.test_isdecimalN)rBrCrDrnrrrrrrrrrrr r"r%r,r2r3r5r7r;r=r?rDrGrrrr r6s2         rc@rz)TestOperationscCrd) Nr rgrhrir|456r}r~rjrmrrr rnHrozTestOperations.setupcCsTtddgddggtj}t||j|jtt|j|jdddkdS)Nabcefg123456789051xyztuvrr) rrrkrlrrrrr=)rABrrr test_addNs $zTestOperations.test_addcCs2tddgddggtj}t|d|jdS)Nqabcq123q789qxyzq)rrrkrlrr)rQArrr test_raddTs zTestOperations.test_raddc Cs|j}dD])}t|d||d|g|d||d|ggtj}t||j|qtdfD]}ttd ||Wdn1sJwYq4dSN)r r<rr*)rr;)r;r)r;r;qrszCan only multiply by integers rrrrkrlrobjectrr&rrrArobrrr test_mulYs  zTestOperations.test_mulc Cs|j}dD])}t|d||d|g|d||d|ggtj}t|||jqtdfD]}ttd ||Wdn1sJwYq4dSrWr[r]rrr test_rmulfs  zTestOperations.test_rmulcCstddgddggtj}tddgddgg}td d gd d ggtj}t|||td dgddggtj}tdd gd d ggtj}t||dtddgd dggtj}t||ddgddggdS)z Ticket #856z%dz%f%s%rr<rXr;rz7.00000019rz%.3fz1.00024r r@N)rrrkrlr)rFr/FCrA1A2rrr test_modrs zTestOperations.test_modc Csxtd|jt|jktd|jt|jkdtfD]}ttd ||jWdn1s4wYqdS)Nrcrd*z*unsupported operand type.* and 'chararray')rrrreprr\rrQ)rr`rrr test_rmods zTestOperations.test_rmodcCstjddgddggddtj}|dd}t||t|j|ut|jj|ju|ddddf}t||t|j|ut|jj|jut|dd kdS) z>Regression test for https://github.com/numpy/numpy/issues/5982rezdef zgeh zijk S4rNr*r)rrrkrlrrbase)rarrsl1sl2rrr test_slices   zTestOperations.test_sliceN) rBrCrDrnrOrVrarbrmrprvrrrr rHGs   rHcCs td}t|gjdkdS)z Regression test for ticket 1948.rvrN)rrlrsize)srrr test_empty_indexings ry)numpyrnumpy.core.multiarrayr numpy.testingrrrrrr'r2r rErcrsr{rrrrrHryrrrr s$ G3j U