o Û¾i›ã@sLddlmZmZddlmZmZddlmZmZddl Z Gdd„deƒZdS)é)ÚTestCaseÚ numpy_support)ÚnjitÚtypes)ÚListÚDictNc@s<eZdZdd„Zdd„Zdd„Zdd„Zd d „Zdd„Zd S)ÚTestConditionsAsPredicatescs°tjtjtjtjtjtjttt t tg}|D]-}dD](}||ƒ‰t‡fdd„ƒ}| |ƒ| ¡¡| |ƒ|dks<|t ur>dnd¡qqtdd„ƒ}d}| ||ƒ| |¡¡dS) N)ércsˆrdSdS©Né é©r ©Úxr ú]/home/ubuntu/.local/lib/python3.10/site-packages/numba/tests/test_conditions_as_predicates.pyÚfooóz4TestConditionsAsPredicates.test_scalars..foor rrcSó|rdSdSr r rr r rrrÚ)ÚnpÚint8Úuint16Úint64Úfloat32Úfloat64Ú complex128ÚintÚfloatÚcomplexÚstrÚboolrÚassertEqualÚpy_func)ÚselfÚdtsÚdtÚcrÚsr rrÚtest_scalars sÿ$ö z'TestConditionsAsPredicates.test_scalarscCsvtdd„ƒ}t tj¡}| ||ƒ| |¡¡| | |¡d¡| d¡| ||ƒ| |¡¡| | |¡d¡dS)NcSrr r rr r rr%rz7TestConditionsAsPredicates.test_typed_list..foorr r)rrÚ empty_listrrr!r"Úappend©r#rÚzr r rÚtest_typed_list$s z*TestConditionsAsPredicates.test_typed_listcCsštdd„ƒ}dg}| ||ƒ| |¡¡| | |¡d¡tdd„ƒ}| |ƒ| ¡¡| | ¡d¡tdd„ƒ}| |ƒ| ¡¡| | ¡d¡dS)NcSrr r rr r rr8rz;TestConditionsAsPredicates.test_reflected_list..foor rcSsddg}|rdSdS©Nr érrr ©Úyr r rrDócSs4ddg}| ¡| ¡t|ƒdksJ‚|rdSdS©Nr r/rrr©ÚpopÚlenr0r r rrPór©rr!r"r+r r rÚtest_reflected_list6ó z.TestConditionsAsPredicates.test_reflected_listcCsštdd„ƒ}dh}| ||ƒ| |¡¡| | |¡d¡tdd„ƒ}| |ƒ| ¡¡| | ¡d¡tdd„ƒ}| |ƒ| ¡¡| | ¡d¡dS)NcSrr r rr r rr`rz:TestConditionsAsPredicates.test_reflected_set..foor rcSsddh}|rdSdSr.r r0r r rrlr2cSs4ddh}| ¡| ¡t|ƒdksJ‚|rdSdSr3r4r0r r rrxr7rr8r+r r rÚtest_reflected_set^r:z-TestConditionsAsPredicates.test_reflected_setcCsxtdd„ƒ}t tjtj¡}| ||ƒ| |¡¡| | |¡d¡d|d<| ||ƒ| |¡¡| | |¡d¡dS)NcSrr r rr r rr‡rz7TestConditionsAsPredicates.test_typed_dict..foorér/r)rrÚemptyrrr!r"r+r r rÚtest_typed_dict†s z*TestConditionsAsPredicates.test_typed_dictcCsÀtdd„ƒ}t d¡}| ||ƒ| |¡¡| | |¡d¡t d¡}| ||ƒ| |¡¡| | |¡d¡t dggg¡}| ||ƒ| |¡¡| | |¡d¡t dggg¡}| ||ƒ| |¡¡| | |¡d¡t d¡}tjdkrŸ| t ¡}||ƒWdƒn1sŽwYd}| |t|jƒ¡n| ||ƒ| |¡¡| | |¡d¡t dd g¡}| t ¡}||ƒWdƒn1sÎwYd }| |t|jƒ¡dS)NcSrr r rr r rr™rz3TestConditionsAsPredicates.test_arrays..foor rrr)r/r/ziThe truth value of an empty array is ambiguous. Use `array.size > 0` to check that an array is not empty.r/z[The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()) rrÚarrayr!r"r=rÚ numpy_versionÚassertRaisesÚ ValueErrorÚassertInrÚ exception)r#rr,ÚraisesÚmsgr r rÚtest_arrays˜s: ÿ ÿz&TestConditionsAsPredicates.test_arraysN) Ú__name__Ú __module__Ú__qualname__r(r-r9r;r>rGr r r rrs((r)Únumba.tests.supportrrÚnumbarrÚnumba.typedrrÚnumpyrrr r r rÚs