o X۷i @sddlmZddlZddlZddlZddlmZddlmZddZddZ d d Z d d Z d dZ ddZ ddZddZdS)) annotationsN)_bundle) _pytest_implcCs4t|tr|jSt|drt|jtr|jSt|S)N__name__) isinstancetyperhasattrstrrepr)objr Q/home/ubuntu/vllm_env/lib/python3.10/site-packages/cupy/testing/_parameterized.py _param_to_str s rcCsfd}t||kr |S|t|d}|t||}|d|||| d}t||ks1J|S)Nz...)len)smaxlenellipsisn1n2r r r _shortens rcsDdd}fddt|D}td||}d|||}|S)Ndics&g|]\}}d|tt|qS)z{}={})formatrr).0kvSINGLE_PARAM_MAXLENr r %sz$_make_class_name..z, z{}_param_{}_{{{}}})sorteditemsrjoinr)base_class_namei_paramparam PARAMS_MAXLEN param_strscls_namer rr _make_class_name!s  r(ccs&t|D] \}}t|||VqdSN) enumerate_parameterize_test_case)baseparamsir$r r r !_parameterize_test_case_generator.sr/csttj|}fdd}d|i}tD]\}ttjr.fdd}|||<q||<q||ddfS)Ncs|}d|fS)Nz<%s parameter: %s>)__repr__)selfname)r,r$r r r08s  z)_parameterize_test_case..__repr__r0csfdd}|S)Ncs|i|Sr)r )r1argskwargsfr r new_vCsz<_parameterize_test_case..create_new_v..new_vr )r7)rr5r create_new_v@s z-_parameterize_test_case..create_new_vcSs|Sr)r )methodr r r Ksz)_parameterize_test_case..)r(rrr rtypes FunctionType)r,r.r$r'r0mbrr8r )r,r$rr r+5s    r+csfdd}|S)aGenerates test classes with given sets of additional attributes >>> @parameterize({"a": 1}, {"b": 2, "c": 3}) ... class TestX(unittest.TestCase): ... def test_y(self): ... pass generates two classes `TestX_param_0_...`, `TestX_param_1_...` and removes the original class `TestX`. The specification is subject to change, which applies to all the non-NumPy `testing` features. csFt|tjrr Jtfdd}||Stji}||S)Ncs t|Sr))r/)r,)r-r r r:as z)parameterize..f..) issubclassunittestTestCasermake_decoratorr parameterize)clsdecor4r-r r r6]s  zparameterize..fr )r-r4r6r rEr rBNsrBcsDttsJtfddD}tj|}fdd|DS)Ncsg|]}|qSr r )rkey) parameterr r rlszproduct..csg|] }tt|qSr )dictzip)rvals)keysr r rns)rrHr itertoolsproduct)rGvaluesvalues_productr )rKrGr rMhs  rMcGsddtj|DS)NcSsg|] }dd|DqS)cSs$i|]}|D]\}}||qqSr )r )rdicrrr r r ts$z+product_dict...r )rdictsr r r rss z product_dict..)rLrM) parametersr r r product_dictqsrT) __future__rrLr;r? cupy.testingrrrrr(r/r+rBrMrTr r r r s