o װi @stddlZddlZddlZddlmZddlmZddZddZdd Zd d Z d d Z ddZ ddZ ddZ dS)N)_bundle) _pytest_implcCs4t|tr|jSt|drt|jtr|jSt|S)N__name__) isinstancetyperhasattrstrrepr)objr O/home/ubuntu/.local/lib/python3.10/site-packages/cupy/testing/_parameterized.py _param_to_str s r cCsfd}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_names  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 r/6s  z)_parameterize_test_case..__repr__r/csfdd}|S)Ncs|i|Sr(r )r0argskwargsfr r new_vAsz<_parameterize_test_case..create_new_v..new_vr )r6)rr4r create_new_v>s z-_parameterize_test_case..create_new_vcSs|Sr(r )methodr r r Isz)_parameterize_test_case..)r'rrrrtypes FunctionType)r+r-r#r&r/mbrr7r )r+r#rr r*3s    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 r9_s z)parameterize..f..) issubclassunittestTestCasermake_decoratorr parameterize)clsdecor3r,r r r5[s  zparameterize..fr )r,r3r5r rDr rALsrAcsDttsJtfddD}tj|}fdd|DS)Ncsg|]}|qSr r )rkey) parameterr r rjszproduct..csg|] }tt|qSr )dictzip)rvals)keysr r rls)rrGr itertoolsproduct)rFvaluesvalues_productr )rJrFr rLfs  rLcGsddtj|DS)NcSsg|] }dd|DqS)cSs$i|]}|D]\}}||qqSr )r)rdicrrr r r rs$z+product_dict...r )rdictsr r r rqs z product_dict..)rKrL) parametersr r r product_dictosrS)rKr:r> cupy.testingrrr rr'r.r*rArLrSr r r r s