B \f@sddlZddlZddlZddlZddlZddlmZdZdZdZ dZ ddZ d d Z d d Z d dZdddZddZddZddZddZdS)N)wrapsz1.0.0z%valuesz %file_pathz%unpackcCst|td|S)z2 Method decorator to add unpack feature. T)setattr UNPACK_ATTR)funcr0lib/python3.7/site-packages/numba/testing/ddt.pyunpacks rcsfdd}|S)z Method decorator to add to your test methods. Should be added to methods of instances of ``unittest.TestCase``. cst|t|S)N)r DATA_ATTR)r)valuesrrwrapper*s zdata..wrapperr)r r r)r rdata#s r csfdd}|S)ac Method decorator to add to your test methods. Should be added to methods of instances of ``unittest.TestCase``. ``value`` should be a path relative to the directory of the file containing the decorated ``unittest.TestCase``. The file should contain JSON encoded data, that can either be a list or a dict. In case of a list, each value in the list will correspond to one test case, and the value will be concatenated to the test method name. In case of a dict, keys will be used as suffixes to the name of the test case, and values will be fed as test data. cst|t|S)N)r FILE_ATTR)r)valuerrr Cs zfile_data..wrapperr)rr r)rr file_data0s rcCs0tjdkrtjdkstjdko.tjjo.dtjkS)NiiiZPYTHONHASHSEED)sys hexversionflagshash_randomizationosenvironrrrris_hash_randomizedIs    rcstdttttfytf7Wntk r4YnXfddtrb|sbd||dSy t|}Wn t k r| dd}YnXd||d|}t d d |S) a Generate a new name for a test case. It will take the original test name and append an ordinal index and a string representation of the value, and convert the result into a valid python identifier by replacing extraneous characters with ``_``. If hash randomization is enabled (a feature available since 2.7.3/3.2.3 and enabled by default since 3.3) and a "non-trivial" value is passed this will omit the name argument by default. Set `PYTHONHASHSEED` to a fixed value before running tests in these cases to get the names back consistently or use the `__name__` attribute on data values. A "trivial" value is a plain scalar, or a tuple or list consisting only of trivial values. Ncs.t|rdSt|ttfr*tt|SdS)NTF) isinstancelisttupleallmap)r) is_trivial trivial_typesrrrvs  z mk_test_name..is_trivialz{0}_{1}asciibackslashreplacez {0}_{1}_{2}z \W|^(?=\d)_) typeboolstrintfloatZunicode NameErrorrformatUnicodeEncodeErrorencoderesub)namerindex test_namer)rrr mk_test_nameQs  r0c sVtfdd}||_jrRyjj|_Wnttfk rPYnX|S)zO This internal method decorator feeds the test data item to the test. cs|fS)Nr)self)argsrkwargsrrr szfeed_data..wrapper)r__name____doc__r( IndexErrorKeyError)rnew_namer2r3r r)r2rr3r feed_datasr9cOst||t||f||dS)z~ Add a test case to this class. The test will be based on an existing function but will give it a new name. N)rr9)clsr/rr2r3rrradd_testsr;c stjt|}tjtj|}fdd}tj|dkr^t|d}t |||dnt t | }x~t|D]r\} } t|tr| || } } t|| | }nt|tr| } t|| | }t| trt |||f| qzt |||| qzWdS)z> Process the parameter in the `file_data` decorator. cstddS)Nz%s does not exist) ValueError)r2) file_attrrr _raise_vesz$process_file_data.._raise_veFerrorN)rpathabspathinspectZ getsourcefilejoindirnameexistsr0r;jsonloadsopenread enumeraterdictr) r:r-rr=Zcls_pathZdata_file_pathr>r/r ielemkeyrr)r=rprocess_file_datas"      rOcCsxt|jD]\}}t|trxtt|tD]n\}}t|t|d||}t|trt |t slt |trt |||f|qt |||f|q2t ||||q2Wt ||qt|t rt|t }t||||t ||qW|S)a Class decorator for subclasses of ``unittest.TestCase``. Apply this decorator to the test case class, and then decorate test methods with ``@data``. For each method decorated with ``@data``, this will effectively create as many methods as data items are passed as parameters to ``@data``. The names of the test methods follow the pattern ``original_test_name_{ordinal}_{data}``. ``ordinal`` is the position of the data argument, starting with 1. For data we use a string representation of the data value converted into a valid python identifier. If ``data.__name__`` exists, we use that instead. For each method decorated with ``@file_data('test_data.json')``, the decorator will try to load the test_data.json file located relative to the python file containing the method that is decorated. It will, for each ``test_name`` key create as many methods in the list of values from the ``data`` key. r4)r__dict__itemshasattrr rJgetattrr0rrrr;delattrr rO)r:r-rrLvr/r=rrrddts     rV)r)rBrFrr+r functoolsr __version__r r rrr rrr0r9r;rOrVrrrrs$    :