o 8ήc @sHddlmZddlmZmZddlmZmZddlZ GdddeZ dS))TestCase)njittypes)ListDictNc@s<eZdZddZddZddZddZd d Zd d Zd S)TestConditionsAsPredicatesc stjtjtjtjtjtjttt t t g }|D]-}dD](}||t fdd}| ||| ||dks<|t ur>dndqqt dd}d}| ||||dS) N)rcs rdSdSN r xr P/tmp/pip-target-vg8gfxp4/lib/python/numba/tests/test_conditions_as_predicates.pyfooz4TestConditionsAsPredicates.test_scalars..foorr r cS |rdSdSr r r r r rrr)npint8uint16int64float32float64 complex128intfloatcomplexstrboolr assertEqualpy_func)selfdtsdtcrsr r r test_scalars s$ z'TestConditionsAsPredicates.test_scalarscCsvtdd}ttj}||||||||d|d||||||||ddS)NcSrr r r r r rr%rz7TestConditionsAsPredicates.test_typed_list..foor rr )rr empty_listrrr r!appendr"rzr r rtest_typed_list$s   z*TestConditionsAsPredicates.test_typed_listcCstdd}dg}||||||||dtdd}|||||dtdd}|||||ddS)NcSrr r r r r rr8rz;TestConditionsAsPredicates.test_reflected_list..foorr cSsddg}|rdSdSNrr r r yr r rrDcSs4ddg}||t|dksJ|rdSdSNrr.rr r poplenr/r r rrPr rr r!r*r r rtest_reflected_list6    z.TestConditionsAsPredicates.test_reflected_listcCstdd}dh}||||||||dtdd}|||||dtdd}|||||ddS)NcSrr r r r r rr`rz:TestConditionsAsPredicates.test_reflected_set..foorr cSsddh}|rdSdSr-r r/r r rrlr1cSs4ddh}||t|dksJ|rdSdSr2r3r/r r rrxr6r r7r*r r rtest_reflected_set^r9z-TestConditionsAsPredicates.test_reflected_setcCsxtdd}ttjtj}||||||||dd|d<||||||||ddS)NcSrr r r r r rrrz7TestConditionsAsPredicates.test_typed_dict..foor r.r )rremptyrrr r!r*r r rtest_typed_dicts z*TestConditionsAsPredicates.test_typed_dictcCsltdd}td}||||||||dtd}||||||||dtdggg}||||||||dtdggg}||||||||dtd}||||||||dtddg}|t }||Wdn1swYd}||t |j dS) NcSrr r r r r rrrz3TestConditionsAsPredicates.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()) rrarrayr r!r< assertRaises ValueErrorassertInr exception)r"rr+raisesmsgr r r test_arrayss.      z&TestConditionsAsPredicates.test_arraysN) __name__ __module__ __qualname__r'r,r8r:r=rEr r r rrs(( r) numba.tests.supportrnumbarr numba.typedrrnumpyrrr r r rs