B <ÓFdí±ã@sddlmZmZmZddlZddlZddlZddlZddlZ ddl mZmZm Z mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZddl Z Gdd„de!ƒZ"Gdd„de"e j#ƒZ$Gdd „d e j#ƒZ%Gd d„de$ƒZ&Gdd „d e"e j#ƒZ'Gdd„de"e j#ƒZ(Gdd„de j#ƒZ)Gdd„de j#ƒZ*Gdd„de j#ƒZ+Gdd„de j#ƒZ,Gdd„de j#ƒZ-Gdd„de j#ƒZ.Gdd„de j#ƒZ/Gdd„de j#ƒZ0d9d d!„Z1d"d#„Z2d$d%„Z3d&d'„Z4d(d)„Z5d*d+„Z6d,d-„Z7d.d/„Z8d0d1„Z9d2d3„Z:Gd4d5„d5eƒZ;d6d7„Zsz_GenericTest.test_objarraycCs| dddgd¡dS)Nr$r%r*)r$r%r*)r!)rrrr Útest_array_likesCsz_GenericTest.test_array_likesN)Ú__name__Ú __module__Ú__qualname__r!r#r(r)r,r-r0r1rrrr rsrc@s<eZdZdd„Zdd„Zdd„Zdd„Zd d „Zdd„Zd S)ÚTestArrayEqualcCs t|_dS)N)rr)rrrr ÚsetUpIszTestArrayEqual.setUpcs<‡fdd„}xdD]}||ƒqWxdD]}||ƒq(WdS)z!Test rank 1 array for all dtypes.csLt d|¡}| d¡| ¡}| ¡}| d¡ˆ ||¡ˆ ||¡dS)Nr%r$r)r&ÚemptyÚfillÚcopyr!r#)ÚtrrÚc)rrr ÚfooNs z.TestArrayEqual.test_generic_rank1..fooz?bhilqpBHILQPfdgFDG)ÚS1ÚU1Nr)rr<r:r)rr Útest_generic_rank1Ls z!TestArrayEqual.test_generic_rank1cs<‡fdd„}xdD]}||ƒqWxdD]}||ƒq(WdS)z!Test rank 3 array for all dtypes.csLt d|¡}| d¡| ¡}| ¡}| d¡ˆ ||¡ˆ ||¡dS)N)r+r%r*r$r)r&r7r8r9r!r#)r:rrr;)rrr r<as z.TestArrayEqual.test_generic_rank3..fooz?bhilqpBHILQPfdgFDG)r=r>Nr)rr<r:r)rr Útest_generic_rank3_s z!TestArrayEqual.test_generic_rank3cCsPt ddtjg¡}t ddtjg¡}| ||¡t dddg¡}| ||¡dS)z$Test arrays with nan values in them.r$r%r*N)r&r'Únanr!r#)rrrr;rrr Útest_nan_arrayrs zTestArrayEqual.test_nan_arraycCsFt ddg¡}t ddg¡}| ||¡t ddg¡}| ||¡dS)z:Test two arrays with different shapes are found not equal.ÚfloupiÚfloupaÚfloupipiN)r&r'r!r#)rrrr;rrr Útest_string_arrays|s z!TestArrayEqual.test_string_arraysc CsÄt ddtjfdtjfg¡}ddg|d<ddg|d<| ¡}| ||¡t ddtjfdtjfg¡}|d ¡|d<|d ¡|d<tƒ0}|jtdd}| ||¡t t |ƒdkƒWdQRXdS) zTest record arrays.r%rCrDr$rEzelementwise == )ÚmessageN)r&r7Úfloatr9r!rÚrecordÚ FutureWarningr#rÚlen)rrrr;ÚsupÚlrrr Útest_recarrays‡szTestArrayEqual.test_recarraysN) r2r3r4r6r?r@rBrFrNrrrr r5Gs r5c@s,eZdZdd„Zdd„Zdd„Zdd„Zd S) ÚTestBuildErrorMessagecCsFt dddg¡}t dddg¡}d}t||g|ƒ}d}| ||¡dS) NgrÄZ| ð?grÄZ| @g«&ˆº@gãˆµøð?g«&ˆº@gãˆµø@zThere is a mismatchzˆ Items are not equal: There is a mismatch ACTUAL: array([ 1.00001, 2.00002, 3.00003]) DESIRED: array([ 1.00002, 2.00003, 3.00004]))r&r'r ÚassertEqual)rÚxÚyÚerr_msgrrrrr Útest_build_err_msg_defaultsœsz1TestBuildErrorMessage.test_build_err_msg_defaultscCsJt dddg¡}t dddg¡}d}t||g|dd }d }| ||¡dS)NgrÄZ| ð?grÄZ| @g«&ˆº@gãˆµøð?g«&ˆº@gãˆµø@zThere is a mismatchF)Úverbosez) Items are not equal: There is a mismatch)r&r'r rP)rrQrRrSrrrrr Útest_build_err_msg_no_verbose§sz3TestBuildErrorMessage.test_build_err_msg_no_verbosecCsJt dddg¡}t dddg¡}d}t||g|dd }d }| ||¡dS)NgrÄZ| ð?grÄZ| @g«&ˆº@gãˆµøð?g«&ˆº@gãˆµø@zThere is a mismatch)ZFOOZBAR)Únamesz Items are not equal: There is a mismatch FOO: array([ 1.00001, 2.00002, 3.00003]) BAR: array([ 1.00002, 2.00003, 3.00004]))r&r'r rP)rrQrRrSrrrrr Útest_build_err_msg_custom_names°sz5TestBuildErrorMessage.test_build_err_msg_custom_namescCsJt dddg¡}t dddg¡}d}t||g|dd }d }| ||¡dS)Ng0¸Dð?grÄZ| @g«&ˆº@g_p‰ð?g«&ˆº@gãˆµø@zThere is a mismatché )Z precisionz Items are not equal: There is a mismatch ACTUAL: array([ 1.000000001, 2.00002 , 3.00003 ]) DESIRED: array([ 1.000000002, 2.00003 , 3.00004 ]))r&r'r rP)rrQrRrSrrrrr Ú#test_build_err_msg_custom_precision»sz9TestBuildErrorMessage.test_build_err_msg_custom_precisionN)r2r3r4rTrVrXrZrrrr rOšs rOc@sTeZdZdd„Zdd„Zdd„Zdd„Zd d „Zdd„Zd d„Z dd„Z dd„ZdS)Ú TestEqualcCs t|_dS)N)rr)rrrr r6ÉszTestEqual.setUpcCsH| tjtj¡| tjgtjg¡| tjtjg¡| tjd¡dS)Nr$)rr&rAr#)rrrr Útest_nan_itemsÌszTestEqual.test_nan_itemscCs:| tjtj¡| tjgtjg¡| tjtjg¡dS)N)rr&Úinfr#)rrrr Útest_inf_itemsÒszTestEqual.test_inf_itemscCs€t d¡}t dd¡}t dd¡}t d¡}t dd¡}t dd¡}|||g}|||g}xBt ||¡D]2\} } | | | ¡| | g| g¡| | g| ¡qfWxBt ||¡D]2\} } | | | ¡| | g| g¡| | g| ¡qªWxšt ||¡D]Š\} } | | | ¡| | | g¡| | g| g¡| | gt dd¡¡| | gt dd¡¡| | gt dd¡¡| | gt dd¡¡qîWdS)NZNaTÚsÚnsz 2017-01-01é{)r&Z datetime64Ztimedelta64Ú itertoolsÚproductrr#)rZnadt_no_unitZnadt_sZnadt_dZnatd_no_unitZnatd_sZnatd_dZdtsZtdsrrrrr Útest_nat_items×s0 zTestEqual.test_nat_itemscCs| dd¡| dd¡dS)NZabZabb)rr#)rrrr Útest_non_numericöszTestEqual.test_non_numericcCsŒ| tddƒtddƒ¡| tdtjƒtdtjƒ¡| tdtjƒtddƒ¡| ttjdƒtdtjƒ¡| ttjtjƒttjdƒ¡dS)Nr$r%)rÚcomplexr&rAr#r])rrrr Útest_complex_itemús zTestEqual.test_complex_itemcCs| tjtj¡dS)N)r#r&ÚPZEROÚNZERO)rrrr Útest_negative_zeroszTestEqual.test_negative_zerocCsRt tddƒtdtjƒg¡}t tddƒtddƒg¡}| ||¡| ||¡dS)Nr$r%)r&r'rfrArr#)rrQrRrrr Útest_complexszTestEqual.test_complexc Cs\y$| t ddg¡t ddg¡¡Wn2tk rV}z| t|ƒd¡Wdd}~XYnXdS)Nr$r%zÃ Arrays are not equal (shapes (2,), (1, 2) mismatch) x: array([1, 2]) y: [repr failed for : The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()])rr&r'Úmatrixr"rPÚstr)rÚerrr Útest_error_message s$zTestEqual.test_error_messageN)r2r3r4r6r\r^rdrergrjrkrorrrr r[Çsr[c@sLeZdZdd„Zdd„Zdd„Zdd„Zd d „Zdd„Zd d„Z dd„Z dS)ÚTestArrayAlmostEqualcCs t|_dS)N)rr)rrrr r6szTestArrayAlmostEqual.setUpcsPˆjddddˆ t‡fdd„¡ˆjdgdgddˆ t‡fdd„¡dS)Ngõ…óþÿ÷?gr)ÚdecimalcsˆjddddS)Ngø?gr)rq)rr)rrr Ú'óz5TestArrayAlmostEqual.test_closeness..csˆjdgdgddS)Ngø?gr)rq)rr)rrr rr,rs)rÚassertRaisesr")rr)rr Útest_closenesssz#TestArrayAlmostEqual.test_closenesscsTt dg¡‰t dg¡‰ˆjˆˆddˆjˆˆddˆ t‡‡‡fdd„¡dS)Ng¯”eˆãH“@gšwœ¢ãH“@r*)rqr+csˆjˆˆddS)Né)rq)rr)rrQrRrr rr5rsz2TestArrayAlmostEqual.test_simple..)r&r'rrtr")rr)rrQrRr Útest_simple.sz TestArrayAlmostEqual.test_simplecs€t tjg¡‰t dg¡‰t tjg¡‰ˆ ˆˆ¡ˆ t‡‡‡fdd„¡ˆ t‡‡‡fdd„¡ˆ t‡‡‡fdd„¡dS)Nr$csˆ ˆˆ¡S)N)rr)ÚananÚaonerrr rr=rsz/TestArrayAlmostEqual.test_nan..csˆ ˆˆ¡S)N)rr)Úainfrxrrr rr?rscsˆ ˆˆ¡S)N)rr)rzrxrrr rrArs)r&r'rAr]rrtr")rr)rzrxryrr Útest_nan7szTestArrayAlmostEqual.test_nancsht ddgddgg¡‰ˆ ¡‰tjˆd<ˆ t‡‡‡fdd„¡tjˆd<ˆ t‡‡‡fdd„¡dS) Ngð?g@g@g@)rrcsˆ ˆˆ¡S)N)rr)rrrrr rrHrsz/TestArrayAlmostEqual.test_inf..csˆ ˆˆ¡S)N)rr)rrrrr rrKrs)r&r'r9r]rtr")rr)rrrr Útest_infCs zTestArrayAlmostEqual.test_infcCsdt ddgddgg¡}tj ddgddggddgddgg¡}| ||¡| ||¡| ||¡dS)Ngð?g@g@g@gFT)r&r'ZmaZmasked_arrayr)rrrrrr Ú test_subclassMsz"TestArrayAlmostEqual.test_subclasscCs®t ddgg¡}t dtjgg¡}t dtjgg¡}t tjtjgg¡}t ddgtjtjgg¡}xD|||||fD]2}| ||¡t |¡}| ||¡| ||¡qtWdS)Ngð?g@)r&rlrAr]rr')rÚm1Úm2Úm3Úm4Úm5Úmrrrr Útest_matrixUs z TestArrayAlmostEqual.test_matrixcs:G‡fdd„dtjƒ‰t ddg¡ ˆ¡}| ||¡dS)Ncs&eZdZ‡‡fdd„Zdd„Z‡ZS)zGTestArrayAlmostEqual.test_subclass_that_cannot_be_bool..MyArraycstˆ|ƒ |¡ tj¡S)N)ÚsuperÚ__lt__Úviewr&Úndarray)rÚother)ÚMyArrayÚ __class__rr r†iszNTestArrayAlmostEqual.test_subclass_that_cannot_be_bool..MyArray.__lt__c_st‚dS)N)ÚNotImplementedError)rÚargsÚkwargsrrr ÚalllszKTestArrayAlmostEqual.test_subclass_that_cannot_be_bool..MyArray.all)r2r3r4r†rÚ __classcell__r)rŠ)r‹r rŠhsrŠgð?g@)r&rˆr'r‡r)rrr)rŠr Ú!test_subclass_that_cannot_be_boolcsz6TestArrayAlmostEqual.test_subclass_that_cannot_be_boolN)r2r3r4r6rurwr{r|r}r„r‘rrrr rps rpc@s\eZdZdd„Zdd„Zdd„Zdd„Zd d „Zdd„Zd d„Z dd„Z dd„Zdd„ZdS)ÚTestAlmostEqualcCs t|_dS)N)rr)rrrr r6uszTestAlmostEqual.setUpcsPˆjddddˆ t‡fdd„¡ˆjdgdgddˆ t‡fdd„¡dS)Ngõ…óþÿ÷?gr)rqcsˆjddddS)Ngø?gr)rq)rr)rrr rr‚rsz0TestAlmostEqual.test_closeness..csˆjdgdgddS)Ngø?gr)rq)rr)rrr rr‡rs)rrtr")rr)rr ruxszTestAlmostEqual.test_closenesscsPˆ tjtj¡ˆ t‡fdd„¡ˆ t‡fdd„¡ˆ t‡fdd„¡dS)Ncsˆ tjd¡S)Nr$)rr&rAr)rrr rrŒrsz/TestAlmostEqual.test_nan_item..csˆ tjtj¡S)N)rr&rAr]r)rrr rrŽrscsˆ tjtj¡S)N)rr&r]rAr)rrr rrrs)rr&rArtr")rr)rr Ú test_nan_item‰szTestAlmostEqual.test_nan_itemcsPˆ tjtj¡ˆ tjtj¡ˆ t‡fdd„¡ˆ t‡fdd„¡dS)Ncsˆ tjd¡S)Nr$)rr&r]r)rrr rr–rsz/TestAlmostEqual.test_inf_item..csˆ tjtj¡S)N)rr&r]r)rrr rr˜rs)rr&r]rtr")rr)rr Ú test_inf_item’szTestAlmostEqual.test_inf_itemcCs| dd¡dS)Nr$r%)r#)rrrr Útest_simple_itemšsz TestAlmostEqual.test_simple_itemcCs¬| tddƒtddƒ¡| tdtjƒtdtjƒ¡| ttjtjƒttjtjƒ¡| tdtjƒtddƒ¡| ttjdƒtdtjƒ¡| ttjtjƒttjdƒ¡dS)Nr$r%)rrfr&rAr]r#)rrrr rgs z!TestAlmostEqual.test_complex_itemcCszt tddƒtdtjƒg¡}t tddƒttjdƒg¡}t tddƒtddƒg¡}| ||¡| ||¡| ||¡dS)Nr$r%)r&r'rfrArr#)rrQÚzrRrrr rk¥szTestAlmostEqual.test_complexc CsÌt dddg¡}t dddg¡}d}y|j||dd Wn<tk rt}z| t|ƒ d ¡d|¡Wdd}~XYnXd }y| ||¡Wn<tk rÆ}z| t|ƒ d ¡d|¡Wdd}~XYnXdS)z>Check the message is formatted correctly for the decimal valuegì¯ð?gì¯@g«&ˆº@gØ_ð?gâ@gãˆµø@zvx: array([ 1.00000000001, 2.00000000002, 3.00003 ]) y: array([ 1.00000000002, 2.00000000003, 3.00004 ])é)rqz%) r$NzRx: array([ 1. , 2. , 3.00003]) y: array([ 1. , 2. , 3.00004]))r&r'rr"rPrmÚsplit)rrQrRrrnrrr ros,z"TestAlmostEqual.test_error_messagecCs®t ddgg¡}t dtjgg¡}t dtjgg¡}t tjtjgg¡}t ddgtjtjgg¡}xD|||||fD]2}| ||¡t |¡}| ||¡| ||¡qtWdS)Ngð?g@)r&rlrAr]rr')rr~rr€rr‚rƒrrrr r„Æs zTestAlmostEqual.test_matrixcs:G‡fdd„dtjƒ‰t ddg¡ ˆ¡}| ||¡dS)Ncs&eZdZ‡‡fdd„Zdd„Z‡ZS)zBTestAlmostEqual.test_subclass_that_cannot_be_bool..MyArraycstˆ|ƒ |¡ tj¡S)N)r…r†r‡r&rˆ)rr‰)rŠr‹rr r†ÚszITestAlmostEqual.test_subclass_that_cannot_be_bool..MyArray.__lt__c_st‚dS)N)rŒ)rrrŽrrr rÝszFTestAlmostEqual.test_subclass_that_cannot_be_bool..MyArray.all)r2r3r4r†rrr)rŠ)r‹r rŠÙsrŠgð?g@)r&rˆr'r‡r)rrr)rŠr r‘Ôsz1TestAlmostEqual.test_subclass_that_cannot_be_boolN) r2r3r4r6rur“r”r•rgrkror„r‘rrrr r’ss r’c@s4eZdZdd„Zdd„Zdd„Zdd„Zd d „ZdS)ÚTestApproxEqualcCs t|_dS)N)rr)rrrr r6æszTestApproxEqual.setUpcsTt dg¡‰t dg¡‰ˆjˆˆddˆjˆˆddˆ t‡‡‡fdd„¡dS)Ng{®GáH“@gR¸…ëH“@rv)ÚsignificantécsˆjˆˆddS)Né)rš)rr)rrQrRrr rrðrsz4TestApproxEqual.test_simple_arrays..)r&r'rrtr")rr)rrQrRr Útest_simple_arraysész"TestApproxEqual.test_simple_arrayscsTd‰d‰ˆjˆˆddˆjˆˆddˆjˆˆddˆ t‡‡‡fdd„¡dS) Ng{®GáH“@gR¸…ëH“@r+)ršrvr›csˆjˆˆddS)Nrœ)rš)rr)rrQrRrr rrúrsz3TestApproxEqual.test_simple_items..)rrtr")rr)rrQrRr Útest_simple_itemsòsz!TestApproxEqual.test_simple_itemscszt tj¡‰t d¡‰t tj¡‰ˆ ˆˆ¡ˆ t‡‡‡fdd„¡ˆ t‡‡‡fdd„¡ˆ t‡‡‡fdd„¡dS)Nr$csˆ ˆˆ¡S)N)rr)rxryrrr rrrsz0TestApproxEqual.test_nan_array..csˆ ˆˆ¡S)N)rr)rzrxrrr rrrscsˆ ˆˆ¡S)N)rr)rzrxrrr rrrs)r&r'rAr]rrtr")rr)rzrxryrr rBüs zTestApproxEqual.test_nan_arraycszt tj¡‰t d¡‰t tj¡‰ˆ ˆˆ¡ˆ t‡‡‡fdd„¡ˆ t‡‡‡fdd„¡ˆ t‡‡‡fdd„¡dS)Nr$csˆ ˆˆ¡S)N)rr)rxryrrr rrrsz0TestApproxEqual.test_nan_items..csˆ ˆˆ¡S)N)rr)rzrxrrr rrrscsˆ ˆˆ¡S)N)rr)rzrxrrr rrrs)r&r'rAr]rrtr")rr)rzrxryrr r\s zTestApproxEqual.test_nan_itemsN)r2r3r4r6rržrBr\rrrr r™äs r™c@sTeZdZdd„Zdd„Zdd„Zdd„Zd d „Zdd„Zd d„Z dd„Z dd„ZdS)ÚTestArrayAssertLesscCs t|_dS)N)r r)rrrr r6szTestArrayAssertLess.setUpcs‚t ddg¡‰t ddg¡‰ˆ ˆˆ¡ˆ t‡‡‡fdd„¡t ddg¡‰ˆ t‡‡‡fdd„¡ˆ t‡‡‡fd d„¡dS) Ngš™™™™™ñ?gš™™™™™@g333333ó?gffffff@csˆ ˆˆ¡S)N)rr)rrQrRrr rr rsz8TestArrayAssertLess.test_simple_arrays..gð?csˆ ˆˆ¡S)N)rr)rrQrRrr rr%rscsˆ ˆˆ¡S)N)rr)rrQrRrr rr'rs)r&r'rrtr")rr)rrQrRr rsz&TestArrayAssertLess.test_simple_arrayscsšt ddgddgg¡‰t ddgddgg¡‰ˆ ˆˆ¡ˆ t‡‡‡fd d „¡t ddgddgg¡‰ˆ t‡‡‡fdd „¡ˆ t‡‡‡fd d „¡dS)Ngš™™™™™ñ?gš™™™™™@gffffff @gš™™™™™@g333333ó?gffffff@g333333@g@csˆ ˆˆ¡S)N)rr)rrQrRrr rr/rsz0TestArrayAssertLess.test_rank2..gð?csˆ ˆˆ¡S)N)rr)rrQrRrr rr4rscsˆ ˆˆ¡S)N)rr)rrQrRrr rr6rs)r&r'rrtr")rr)rrQrRr Ú test_rank2)szTestArrayAssertLess.test_rank2cs|tjdd‰tjddd‰ˆ ˆˆ¡ˆ t‡‡‡fdd„¡dˆd<ˆ t‡‡‡fdd„¡ˆ t‡‡‡fd d„¡dS) N)r%r%r%)Úshaper$csˆ ˆˆ¡S)N)rr)rrQrRrr rr>rsz0TestArrayAssertLess.test_rank3..r)rrrcsˆ ˆˆ¡S)N)rr)rrQrRrr rrCrscsˆ ˆˆ¡S)N)rr)rrQrRrr rrErs)r&Úonesrrtr")rr)rrQrRr Ú test_rank38szTestArrayAssertLess.test_rank3csˆd‰d‰ˆ ˆˆ¡ˆ t‡‡‡fdd„¡t ddg¡‰ˆ ˆˆ¡ˆ t‡‡‡fdd„¡t ddg¡‰ˆ t‡‡‡fdd„¡dS) Ngš™™™™™ñ?gš™™™™™@csˆ ˆˆ¡S)N)rr)rrQrRrr rrMrsz7TestArrayAssertLess.test_simple_items..gffffff @csˆ ˆˆ¡S)N)rr)rrQrRrr rrSrsgð?csˆ ˆˆ¡S)N)rr)rrQrRrr rrXrs)rrtr"r&r')rr)rrQrRr ržGsz%TestArrayAssertLess.test_simple_itemscs’t tj¡‰t d¡‰t tj¡‰ˆ ˆˆ¡ˆ t‡‡‡fdd„¡ˆ t‡‡‡fdd„¡ˆ t‡‡‡fdd„¡ˆ t‡‡‡fdd„¡dS)Nr$csˆ ˆˆ¡S)N)rr)rxryrrr rr`rsz9TestArrayAssertLess.test_nan_noncompare..csˆ ˆˆ¡S)N)rr)rxryrrr rrbrscsˆ ˆˆ¡S)N)rr)rzrxrrr rrdrscsˆ ˆˆ¡S)N)rr)rzrxrrr rrfrs)r&r'rAr]rrtr")rr)rzrxryrr Útest_nan_noncompareZs z'TestArrayAssertLess.test_nan_noncomparecsÈt dddg¡‰t tj¡‰ˆ t‡‡‡fdd„¡ˆ t‡‡‡fdd„¡t ddtjg¡‰ˆ t‡‡‡fdd„¡ˆ t‡‡‡fdd„¡t d d tjg¡‰ˆ ˆˆ¡ˆ t‡‡‡fdd„¡dS)Ngš™™™™™ñ?gš™™™™™@gffffff @csˆ ˆˆ¡S)N)rr)rxrrQrr rrmrsz?TestArrayAssertLess.test_nan_noncompare_array..csˆ ˆˆ¡S)N)rr)rxrrQrr rrorscsˆ ˆˆ¡S)N)rr)rxrrQrr rrtrscsˆ ˆˆ¡S)N)rr)rxrrQrr rrvrsgð?g@csˆ ˆˆ¡S)N)rr)rrQrRrr rr|rs)r&r'rArtr"r)rr)rxrrQrRr Útest_nan_noncompare_arrayhsz-TestArrayAssertLess.test_nan_noncompare_arraycs´t d¡‰t tj¡‰ˆ ˆˆ¡ˆ ˆˆ¡ˆ ˆˆ¡ˆ t‡‡‡fdd„¡ˆ t‡‡‡fdd„¡ˆ t‡‡fdd„¡ˆ t‡‡fdd„¡ˆ t‡‡fdd„¡dS)Nr$csˆ ˆˆ¡S)N)rr)rzryrrr rr†rsz6TestArrayAssertLess.test_inf_compare..csˆ ˆˆ¡S)N)rr)rzryrrr rrˆrscsˆ ˆˆ¡S)N)rr)rzrrr rrŠrscsˆ ˆˆ¡S)N)rr)rzrrr rrŒrscsˆ ˆˆ¡S)N)rr)rzrrr rrŽrs)r&r'r]rrtr")rr)rzryrr Útest_inf_compare~s z$TestArrayAssertLess.test_inf_comparecs¨t ddtjg¡‰t tj¡‰ˆ t‡‡‡fdd„¡ˆ t‡‡‡fdd„¡ˆ t‡‡‡fdd„¡ˆ t‡‡‡fdd„¡ˆ t‡‡‡fdd„¡ˆ ˆˆ¡dS) Ngš™™™™™ñ?gš™™™™™@csˆ ˆˆ¡S)N)rr)rzrrQrr rr•rsz.csˆ ˆˆ¡S)N)rr)rzrrQrr rr—rscsˆ ˆˆ¡S)N)rr)rzrrQrr rr™rscsˆ ˆˆ¡S)N)rr)rzrrQrr rr›rscsˆ ˆˆ¡S)N)rr)rzrrQrr rrrs)r&r'r]rtr"r)rr)rzrrQr Útest_inf_compare_arraysz*TestArrayAssertLess.test_inf_compare_arrayN)r2r3r4r6rr r£ržr¤r¥r¦r§rrrr rŸsrŸc@s<eZdZdd„Zdd„Zdd„Zdd„Zd d „Zdd„Zd S)Ú TestRaisescCsGdd„dtƒ}||_dS)Nc@seZdZdS)z%TestRaises.setUp..MyExceptionN)r2r3r4rrrr ÚMyException¤sr©)Ú Exceptionrn)rr©rrr r6£szTestRaises.setUpcCs|‚dS)Nr)rrnrrr Úraises_exception©szTestRaises.raises_exceptioncCsdS)Nr)rrrr Údoes_not_raise_exception¬sz#TestRaises.does_not_raise_exceptioncCst|jƒ|jƒ|jƒdS)N)rrnr«)rrrr Útest_correct_catch¯szTestRaises.test_correct_catchcCs:yt|jƒ|jƒtƒWntk r,dSXtdƒ‚dS)Nzshould have caught RuntimeError)rrnr«ÚRuntimeErrorr")rrrr Útest_wrong_exception²s zTestRaises.test_wrong_exceptioncCs8yt|jƒ|jƒƒWntk r*dSXtdƒ‚dS)Nz$should have raised an AssertionError)rrnr¬r")rrrr Útest_catch_no_raiseºs zTestRaises.test_catch_no_raiseN) r2r3r4r6r«r¬rr¯r°rrrr r¨¡sr¨c@s$eZdZdd„Zdd„Zdd„ZdS)Ú TestWarnscCshdd„}tjdjdd…}ttt|ƒdƒtjdj}ttt|ƒttdd„dƒdƒt||dƒdS) NcSst d¡dS)NÚyor*)ÚwarningsÚwarnrrrr ÚfÆs zTestWarns.test_warn..fr³r*cSs|S)Nr)rQrrr rrÏrsz%TestWarns.test_warn..r$z.assert_warns does not preserver warnings state) ÚsysÚmodulesÚfiltersrr ÚUserWarningrr"r)rrµÚbefore_filtersÚ after_filtersrrr Ú test_warnÅszTestWarns.test_warnc Cs`tjdjdd…}ttƒt d¡WdQRXtjdj}dd„}tt|ƒt ||dƒdS)Nr³r²c Ss tƒt d¡WdQRXdS)Nr²)rr³r´rrrr Úno_warningsÜsz3TestWarns.test_context_manager..no_warningsz.assert_warns does not preserver warnings state) r¶r·r¸r r¹r³r´rr"r)rrºr»r½rrr Útest_context_managerÕs zTestWarns.test_context_managerc Csddd„}d}t ¡:t dt¡ytt|ƒd}Wntk rHYnXWdQRX|r`tdƒ‚dS)NcSst dt¡dS)Nr²)r³r´ÚDeprecationWarningrrrr rµåsz,TestWarns.test_warn_wrong_warning..fFÚerrorTz#wrong warning caught by assert_warn)r³Úcatch_warningsÚsimplefilterr¿r r¹r")rrµZfailedrrr Útest_warn_wrong_warningäs z!TestWarns.test_warn_wrong_warningN)r2r3r4r¼r¾rÃrrrr r±Ãsr±c@s<eZdZdd„Zdd„Zdd„Zdd„Zd d „Zdd„Zd S)ÚTestAssertAllclosecCs¸d}d}t||dd| tt||¡t ||||g¡}t ||||g¡}t||dd| tt||¡|d|d<t||ƒ|jtt||ddtdd d d|jttd dd ddS)Ngü©ñÒMbP?g•Ö&è.>r$)ZatolgÜ1¯ð?éÿÿÿÿ)Zrtolr›rYgà?)rrtr"r&r')rrQrRrrrrr rwøs zTestAssertAllclose.test_simplecCs*tjt tj¡jgtjd}t||ƒdS)N)r.)r&r'ZiinfoZint_Úminr)rrrrr Útest_min_int szTestAssertAllclose.test_min_intc Csvt ddddg¡}t ddddg¡}yt||ƒd}Wn,tk rb}z|jd}Wdd}~XYnX| d|k¡dS)Nr$r%Úrzmismatch 25.0%)r&r'rr"rZ assertTrue)rrrÚmsgÚexcrrr Útest_report_fail_percentages z.TestAssertAllclose.test_report_fail_percentagecCs.t tjg¡}t tjg¡}t||dddS)NT)Ú equal_nan)r&r'rAr)rrrrrr Útest_equal_nansz!TestAssertAllclose.test_equal_nancCs4t tjg¡}t tjg¡}|jtt||dddS)NF)rÌ)r&r'rArtr"r)rrrrrr Útest_not_equal_nan"sz%TestAssertAllclose.test_not_equal_nancCsHt tjg¡}t tjg¡}t||ƒt||ƒt||ƒt||ƒdS)N)r&r'rArrr r)rrrrrr Útest_equal_nan_default(s z)TestAssertAllclose.test_equal_nan_defaultN) r2r3r4rwrÇrËrÍrÎrÏrrrr rÄös rÄc@sLeZdZdd„Zdd„Zdd„Zdd„Zd d „Zdd„Zd d„Z dd„Z dS)ÚTestArrayAlmostEqualNulpcCsd}tjdddtjd}d|}tj||f}t |j¡j}||||d}t|||ƒt |j¡j}||||d}t|||ƒdS)Nrviìÿÿÿéé2)r.rYg@) r&ÚlinspaceÚfloat64Úr_Úfinfor.ÚepsrÚepsneg)rÚnulprQr×rRrØrrr Útest_float64_pass6sz*TestArrayAlmostEqualNulp.test_float64_passcCsœd}tjdddtjd}d|}tj||f}t |j¡j}||||d}| tt |||¡t |j¡j }||||d}| tt |||¡dS)NrviìÿÿÿrÑrÒ)r.rYg@)r&rÓrÔrÕrÖr.r×rtr"rrØ)rrÙrQr×rRrØrrr Útest_float64_failHs z*TestArrayAlmostEqualNulp.test_float64_failcCsd}tjdddtjd}d|}tj||f}t |j¡j}||||d}t|||ƒt |j¡j}||||d}t|||ƒdS)NrviìÿÿÿrÑrÒ)r.rYg@) r&rÓÚfloat32rÕrÖr.r×rrØ)rrÙrQr×rRrØrrr Útest_float32_passXsz*TestArrayAlmostEqualNulp.test_float32_passcCsœd}tjdddtjd}d|}tj||f}t |j¡j}||||d}| tt |||¡t |j¡j }||||d}| tt |||¡dS)NrviìÿÿÿrÑrÒ)r.rYg@)r&rÓrÜrÕrÖr.r×rtr"rrØ)rrÙrQr×rRrØrrr Útest_float32_failfs z*TestArrayAlmostEqualNulp.test_float32_failcCs$d}tjdddtjd}d|}tj||f}||d}t |j¡j}||||d}t|||d|ƒt|||d|ƒ||||d }t|||d|ƒt |j¡j}||||d}t|||d|ƒt|||d|ƒ||||d }t|||d|ƒdS) NrviìÿÿÿrÑrÒ)r.rYyð?g@g@) r&rÓrÔrÕrÖr.r×rrØ)rrÙrQÚxir×rRrØrrr Útest_complex128_passvs"z-TestArrayAlmostEqualNulp.test_complex128_passcCs@d}tjdddtjd}d|}tj||f}||d}t |j¡j}||||d}| tt |||d|¡| tt |||d|¡||||}| tt |||d|¡t |j¡j }||||d}| tt |||d|¡| tt |||d|¡||||}| tt |||d|¡dS) NrviìÿÿÿrÑrÒ)r.rYyð?g@)r&rÓrÔrÕrÖr.r×rtr"rrØ)rrÙrQrßr×rRrØrrr Útest_complex128_fails.z-TestArrayAlmostEqualNulp.test_complex128_failcCs$d}tjdddtjd}d|}tj||f}||d}t |j¡j}||||d}t|||d|ƒt|||d|ƒ||||d }t|||d|ƒt |j¡j}||||d}t|||d|ƒt|||d|ƒ||||d }t|||d|ƒdS) NrviìÿÿÿrÑrÒ)r.rYyð?g@g@) r&rÓrÜrÕrÖr.r×rrØ)rrÙrQrßr×rRrØrrr Útest_complex64_passªs"z,TestArrayAlmostEqualNulp.test_complex64_passcCs@d}tjdddtjd}d|}tj||f}||d}t |j¡j}||||d}| tt |||d|¡| tt |||d|¡||||}| tt |||d|¡t |j¡j }||||d}| tt |||d|¡| tt |||d|¡||||}| tt |||d|¡dS) NrviìÿÿÿrÑrÒ)r.rYyð?g@)r&rÓrÜrÕrÖr.r×rtr"rrØ)rrÙrQrßr×rRrØrrr Útest_complex64_fail¿s.z,TestArrayAlmostEqualNulp.test_complex64_failN)r2r3r4rÚrÛrÝrÞràrárârãrrrr rÐ4srÐc@s4eZdZdd„Zdd„Zdd„Zdd„Zd d „ZdS)ÚTestULPcCstj d¡}t||dddS)NrYr)Úmaxulp)r&ÚrandomÚrandnr)rrQrrr Ú test_equalÝszTestULP.test_equalcCsRt d¡ tj¡}|dtj d¡ tj¡7}t tj¡j}t|||dddS)NrYg{®Gáz„?rÑ)rå) r&r¢ÚastyperÜrærçrÖr×r)rrQr×rrr Útest_singleászTestULP.test_singlecCsRt d¡ tj¡}|dtj d¡ tj¡7}t tj¡j}t|||dddS)NrYg{®Gáz„?éÈ)rå) r&r¢rérÔrærçrÖr×r)rrQr×rrr Útest_doubleèszTestULP.test_doublecCsPxJtjtjgD]:}t tjg¡ |¡}t t |¡jg¡}t||ddqWdS)Nrë)rå) r&rÜrÔr'r]rérÖÚmaxr)rÚdtr]Úbigrrr r|ïszTestULP.test_infcs"xtjtjgD]}|tjkr&d‰nd‰t tjg¡ |¡‰t tjg¡ |¡‰t t |¡jg¡‰t t |¡j g¡‰t tj g¡ |¡‰t tjg¡ |¡‰| t ‡‡‡fdd„¡| t ‡‡‡fdd„¡| t ‡‡‡fdd„¡| t ‡‡‡fdd„¡| t ‡‡‡fdd„¡qWdS) Ng€„.Ag¢”mBcstˆˆˆdS)N)rå)rr)r]rårArr rrsz"TestULP.test_nan..cstˆˆˆdS)N)rå)rr)rïrårArr rrscstˆˆˆdS)N)rå)rr)rårAÚtinyrr rr scstˆˆˆdS)N)rå)rr)rårAÚzerorr rrscstˆˆˆdS)N)rå)rr)rårAÚnzerorr rrs)r&rÜrÔr'r]rérArÖrírðrhrirtr")rrîr)rïr]rårAròrðrñr r{õs( zTestULP.test_nanN)r2r3r4rèrêrìr|r{rrrr räÛs räc@seZdZdd„ZdS)ÚTestStringEqualc CsptddƒtddƒytddƒWn0tk rR}ztt|ƒdƒWdd}~XYn Xtdƒ‚| tdd„¡dS) NÚhellozhello multilinezfoo barz hello barz%Differences in strings: - foo + hellozexception not raisedcSs tddƒS)Nr<rô)rrrrr rr rsz-TestStringEqual.test_simple..)rr"rrmrt)rrÊrrr rws zTestStringEqual.test_simpleN)r2r3r4rwrrrr rósrócCs@|j}d|kr.|dkr|}tt|ƒd|ƒntt|ƒ|ƒdS)NÚversionr$)Ú__warningregistry__rrK)ÚmodZn_in_contextZpy3_n_in_contextZ mod_warnsrrr Úassert_warn_len_equal#srøcCs2tjt}y|j ¡Wntk r,YnX|S)N)r¶r·r2röÚclearÚAttributeError)Úmy_modrrr Ú_get_fresh_mod1s rüc Csìtƒ}tt|diƒiƒt|gdt d¡t d¡WdQRXt|jiƒtƒt d¡t d¡WdQRXt|dƒt|gdt d¡t d¡WdQRXt|dƒtƒt d¡t d¡WdQRXt|ddƒdS)Nrö)r·ÚignorezSome warningr$zAnother warningr%) rürÚgetattrrr³rÂr´rörø)rûrrr Útest_clear_and_catch_warnings;s$ rÿc Cs,tƒ}tt|diƒiƒdd„}tƒ0}| t¡|jtjj dt d¡|ƒWdQRXtt |jƒdkƒt|jdjjddkƒt|dƒtƒ}|j|d|t d¡WdQRXt|dƒ|j|d|t d¡WdQRXt|dƒtƒt d¡t d¡WdQRXt|dƒdS) NröcSsdd„}t |ddg¡dS)NcSstjddd|S)NzSome warning 2r%)Ú stacklevel)r³r´)Úarrrrr r´]szFtest_suppress_warnings_module..warn_other_module..warnr)r&Úapply_along_axis)r´rrr Úwarn_other_moduleZsz8test_suppress_warnings_module..warn_other_module)ÚmodulezSome warningr$rrý)rürrþrrIr¹Úfilterr&ÚlibZ shape_baser³r´rrKÚlogrGrrørÂ)rûrrLrrr Útest_suppress_warnings_moduleUs0 rc Csàtƒ}tt|diƒiƒtƒ}| t¡t d¡WdQRXt|dƒtƒ}| t¡|t d¡WdQRXt|dƒ|j|d|t d¡WdQRXt|dƒtƒt d¡t d¡WdQRXt|dƒdS)NrözSome warningr)rrýr$) rürrþrrr¹r³r´rørÂ)rûrLrrr Útest_suppress_warnings_typeƒs& r c Csbtƒ}| t¡|dd„ƒ}tjdd0}t d¡|tƒ|tƒtt|ƒdkƒWdQRXdS)NcSst d|¡dS)NzSome warning)r³r´)Úcategoryrrr r´£sz7test_suppress_warnings_decorate_no_record..warnT)rIÚalwaysr$) rrr¹r³rÁrÂÚRuntimeWarningrrK)rLr´Úwrrr Ú)test_suppress_warnings_decorate_no_recordŸs rc Cs¢tƒ}| ¡}|†|jdd}|jddt d¡t d¡t d¡tt|jƒdkƒtt|ƒdkƒtt|ƒdkƒt|djj ddkƒWdQRX|†|jdd}|jddt d¡t d¡t d¡tt|jƒdkƒtt|ƒdkƒtt|ƒdkƒt|djj ddkƒWdQRXtƒd}| ¡tƒ8}|jddt d¡t d¡tt|jƒdkƒWdQRXtt|jƒdkƒWdQRXdS)NzSome other warning 2)rGzSome warningzSome other warningr%r$r) rrIrr³r´rrKrrGr)rLZlog1Zlog2Zsup2rrr Útest_suppress_warnings_record®s< " " rc Cs¨dd„}tƒP}| ¡tdƒ"xtdƒD]}t d¡q,WWdQRXtt|jƒdkƒWdQRXtƒZ}| ¡tdƒ,x$tdƒD]}t d¡t d¡qˆWWdQRXtt|jƒdkƒWdQRXtƒ`}| ¡tdƒ2x*tdƒD]}t d¡t d¡|ƒqîWWdQRXtt|jƒdkƒWdQRXtƒb}| ¡tdƒ4x,tdƒD] }t d¡t d ¡|ƒqZWWdQRXtt|jƒdkƒWdQRXdS) NcSsdd„}t |ddg¡dS)NcSstjddd|S)NzSome warningr%)r)r³r´)rrrr r´ÚszJtest_suppress_warnings_forwarding..warn_other_module..warnr)r&r)r´rrr r×sz.warn_other_modulerr%zSome warningÚlocationrÚoncezSome other warning)rrIÚranger³r´rrKr)rrLÚirrr Ú!test_suppress_warnings_forwardingÖs< rc Cs tƒ*}tj |d¡}t|dƒWdQRXWdQRXttj |¡ƒd}ytƒ}tƒ‚WdQRXWntk r€d}YnXt|ƒttj |¡ƒdS)NZtmpr FT)rÚosÚpathÚjoinÚopenrÚisdirÚ ValueError)ZtdirÚfpathÚraisedrrr Útest_tempdirs rc Cs’tƒ}t|dƒ}WdQRXWdQRXttj |¡ƒd}ytƒ}tƒ‚WdQRXWntk rrd}YnXt|ƒttj |¡ƒdS)Nr FT)rrrrrÚisfiler)rrµrrrr Ú test_temppaths rc@seZdZejefZdS)Úmy_cacwN)r2r3r4r¶r·Z class_modulesrrrr r &sr c Cs<tƒ}tƒt d¡t d¡WdQRXt|jiƒdS)NrýzSome warning)rür r³rÂr´rrö)rûrrr Ú%test_clear_and_catch_warnings_inherit+s r!Ú__main__)N)>Z __future__rrrr³r¶rrbZnumpyr&Z numpy.testingrrrrr r rrr rrrrrrrrrrrrZunittestr/rZTestCaser5rOr[rpr’r™rŸr¨r±rÄrÐrärórørürÿrr rrrrrr r!r2rrrr ÚsJ\ 2S-Q[q1 "3>(8 .(/