U ,‰d ã@sHddlmZddlmZmZddlmZmZddlZ Gdd„deƒZ dS)é)ÚTestCase)ÚnjitÚtypes)ÚListÚDictNc@s<eZdZdd„Zdd„Zdd„Zdd„Zd d „Zd d „Zd S)ÚTestConditionsAsPredicatesc s°tjtjtjtjtjtjttt t t g }|D]Z}dD]P}||ƒ‰t ‡fdd„ƒ}|  |ƒ| ¡¡|  |ƒ|dksx|t kr|dnd¡q2q*t dd„ƒ}d}|  ||ƒ| |¡¡dS) N)ércsˆrdSdSdS©Né é©r ©Úxr úM/tmp/pip-unpacked-wheel-eu7e0c37/numba/tests/test_conditions_as_predicates.pyÚfoosz4TestConditionsAsPredicates.test_scalars..foorr r cSs|rdSdSdSr r r r r rrsÚ)ÚnpZint8Zuint16Úint64Zfloat32Zfloat64Z complex128ÚintÚfloatÚcomplexÚstrÚboolrÚ assertEqualÚpy_func)ÚselfZdtsÚdtÚcrÚsr r rÚ test_scalars s&ÿ& z'TestConditionsAsPredicates.test_scalarscCsvtdd„ƒ}t tj¡}| ||ƒ| |¡¡| | |¡d¡| d¡| ||ƒ| |¡¡| | |¡d¡dS)NcSs|rdSdSdSr r r r r rr%sz7TestConditionsAsPredicates.test_typed_list..foor rr )rrZ empty_listrrrrÚappend©rrÚzr r rÚtest_typed_list$s   z*TestConditionsAsPredicates.test_typed_listcCsštdd„ƒ}dg}| ||ƒ| |¡¡| | |¡d¡tdd„ƒ}| |ƒ| ¡¡| | ¡d¡tdd„ƒ}| |ƒ| ¡¡| | ¡d¡dS)NcSs|rdSdSdSr r r r r rr8sz;TestConditionsAsPredicates.test_reflected_list..foorr cSsddg}|rdSdSdS©Nrér r r ©Úyr r rrDscSs8ddg}| ¡| ¡t|ƒdks(t‚|r0dSdSdS©Nrr%rr r ©ÚpopÚlenÚAssertionErrorr&r r rrPsr ©rrrr!r r rÚtest_reflected_list6s   z.TestConditionsAsPredicates.test_reflected_listcCsštdd„ƒ}dh}| ||ƒ| |¡¡| | |¡d¡tdd„ƒ}| |ƒ| ¡¡| | ¡d¡tdd„ƒ}| |ƒ| ¡¡| | ¡d¡dS)NcSs|rdSdSdSr r r r r rr`sz:TestConditionsAsPredicates.test_reflected_set..foorr cSsddh}|rdSdSdSr$r r&r r rrlscSs8ddh}| ¡| ¡t|ƒdks(t‚|r0dSdSdSr(r)r&r r rrxsr r-r!r r rÚtest_reflected_set^s   z-TestConditionsAsPredicates.test_reflected_setcCsxtdd„ƒ}t tjtj¡}| ||ƒ| |¡¡| | |¡d¡d|d<| ||ƒ| |¡¡| | |¡d¡dS)NcSs|rdSdSdSr r r r r rr‡sz7TestConditionsAsPredicates.test_typed_dict..foor ér%r )rrÚemptyrrrrr!r r rÚtest_typed_dict†s z*TestConditionsAsPredicates.test_typed_dictc CsXtdd„ƒ}t d¡}| ||ƒ| |¡¡| | |¡d¡t d¡}| ||ƒ| |¡¡| | |¡d¡t dggg¡}| ||ƒ| |¡¡| | |¡d¡t dggg¡}| ||ƒ| |¡¡| | |¡d¡t d¡}| ||ƒ| |¡¡| | |¡d¡t ddg¡}| t¡}||ƒW5QRXd}| |t |j ƒ¡dS) NcSs|rdSdSdSr r r r r rr™sz3TestConditionsAsPredicates.test_arrays..foorr rr r%z[The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()) rrÚarrayrrr1Ú assertRaisesÚ ValueErrorZassertInrÚ exception)rrr"ZraisesÚmsgr r rÚ test_arrays˜s,     z&TestConditionsAsPredicates.test_arraysN) Ú__name__Ú __module__Ú __qualname__rr#r.r/r2r8r r r rrs ((r) Znumba.tests.supportrZnumbarrZ numba.typedrrZnumpyrrr r r rÚs