o ºi›ã@sLddlmZmZddlmZmZddlmZmZddl Z Gdd„deƒZ dS)é)ÚTestCaseÚ numpy_support)ÚnjitÚtypes)ÚListÚDictNc@s<eZdZdd„Zdd„Zdd„Zdd„Zd d „Zd d „Zd S)ÚTestConditionsAsPredicatesc s°tjtjtjtjtjtjttt t t g }|D]-}dD](}||ƒ‰t ‡fdd„ƒ}|  |ƒ| ¡¡|  |ƒ|dks<|t ur>dnd¡qqt dd„ƒ}d}|  ||ƒ| |¡¡dS) N)ércs ˆrdSdS©Né é©r ©Úxr úf/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/numba/tests/test_conditions_as_predicates.pyÚfooóz4TestConditionsAsPredicates.test_scalars..foor r r cSó |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..foor r 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 r cSsddg}|rdSdS©Nr ér r r ©Úyr r rrDócSs4ddg}| ¡| ¡t|ƒdksJ‚|rdSdS©Nr r/rr r ©Ú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 r cSsddh}|rdSdSr.r r0r r rrlr2cSs4ddh}| ¡| ¡t|ƒdksJ‚|rdSdSr3r4r0r r rrxr7r r8r+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 r rr )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