ShdZddlZddlmZmZmZmZmZmZm Z m Z ddl Z ddl m Z ddlmZmZmZdZgdZeeeeZd ded eeeefd eed ffd Z d!ded eeeefd eed eefdZe d"dedededededeedededd eeeeeefffdZe d#dedededededeedededd eeeffdZ d$dedededededeededed e eeeeeeffeeefffdZded eefdZdS)%z Testing routines for opt_einsum.N)AnyDictListLiteralOptionalTupleUnionoverload) get_symbol) ArrayTypePathTypeTensorShapeTypeabcdefghijklmopqABC)rrrrrrrrrrrrrstringdimension_dictreturn.ctg}|ddd}|D]2}fd|D}|t|3t|S)aBuilds random tensor shapes for testing. Parameters: string: List of tensor strings to build dimension_dict: Dictionary of index sizes, defaults to indices size of 2-7 Returns: The resulting shapes. Examples: ```python >>> shapes = build_shapes('abbc', {'a': 2, 'b':3, 'c':5}) >>> shapes [(2, 3), (3, 3, 5), (5,)] ``` Nz->r,c g|] }| Sr).0xrs X/var/www/html/movieo_spanner_bot/venv/lib/python3.11/site-packages/opt_einsum/testing.py z build_shapes..(s000aq!000)_default_dim_dictsplitappendtuple)rrshapestermstermdimss ` r build_shapesr*s$* F LL  q ! ' ' , ,E##00004000 eDkk"""" ==r!array_functionc| tjd}|jj}g}t ||D]C}|r||||tjDt |S)aBuilds random numpy arrays for testing. Parameters: string: List of tensor strings to build dimension_dict: Dictionary of index _sizes array_function: Function to build the arrays, defaults to np.random.rand Returns: The resulting views. Examples: ```python >>> view = build_views('abbc', {'a': 2, 'b':3, 'c':5}) >>> view[0].shape (2, 3, 3, 5) ``` Nnumpy)r)pytest importorskiprandomrandr*r$r%)rrr+npviewsshapes r build_viewsr5-s*   ) ) Ef^DDD**  * LL/ 0 0 0 0 LL ) ) ) ) <<r!n regularityn_outd_mind_maxseed global_dimreturn_size_dictTcdSNrr6r7r8r9r:r;r<r=s r rand_equationrAOs ,/3r!FcdSr?rr@s rrArA]s 3r!r ctjd|j|||zdzz}dt |D} gfdt |Dfd} t jt| D]v\} } | |kr| | xx| z cc<jd|} | | | vr%jd|} | | | v%| | xx| z cc<w|rWt|}jdz|<t |D]} | | xx|z cc<|z d jd d | }fd | D}||f}|r|fzS|S) a9Generate a random contraction and shapes. Parameters: n: Number of array arguments. regularity: 'Regularity' of the contraction graph. This essentially determines how many indices each tensor shares with others on average. n_out: Number of output indices (i.e. the number of non-contracted indices). Defaults to 0, i.e., a contraction resulting in a scalar. d_min: Minimum dimension size. d_max: Maximum dimension size. seed: If not None, seed numpy's random generator with this. global_dim: Add a global, 'broadcast', dimension to every operand. return_size_dict: Return the mapping of indices to sizes. Returns: eq: The equation string. shapes: The array shapes. size_dict: The dict of index sizes, only returned if ``return_size_dict=True``. Examples: ```python >>> eq, shapes = rand_equation(n=10, regularity=4, n_out=5, seed=42) >>> eq 'oyeqn,tmaq,skpo,vg,hxui,n,fwxmr,hitplcj,kudlgfv,rywjsb->cebda' >>> shapes [(9, 5, 4, 5, 4), (4, 4, 8, 5), (9, 4, 6, 9), (6, 6), (6, 9, 7, 8), (4,), (9, 3, 9, 4, 9), (6, 8, 4, 6, 8, 6, 3), (4, 7, 8, 8, 6, 9, 6), (9, 5, 3, 3, 9, 5)] ``` r-Nrcg|]}dS)r)r_s rr z!rand_equation..s # # #Qb # # #r!chi|].}t|jdz/S))r r0randint)rir:r9r2s r z!rand_equation..s7]]]A 1 1% C C]]]r!c3KtD]-\}}|kr||V%|V|V.dSr?) enumerater$)rKixr8output size_dicts rgenzrand_equation..genshy))  EAr5yy b!!!  r!rrIrFz{}->{}rcFg|]}tfd|DS)c3(K|] }|V dSr?r)rrOrQs r z+rand_equation...s'//bIbM//////r!)r%)roprQs rr z!rand_equation..s6 A A ABe////B///// A A Ar!) r.r/r0r;rangerN permutationlistrJr joinformat)r6r7r8r9r:r;r<r=num_indsinputsrRrKrOwheregdimeqr&retr2rPrQs ``` @@@rrArAjsa`  W % %B  t:~"U*H # #%(( # # #F F]]]]]]US[__]]]I       2900ccee==>>   2 q55 1IIIOIIIII%%a++Eu %% ))!Q//u %% 5MMMR MMMM(##)++E519== $q  A 1III IIII$WWRY**622 3 3F &))6 2 2BB A A A& A A AF v,Ci\!! r!pathcHtjdfd|DS)zBuild random numpy arrays from a path. Parameters: path: The path to build arrays from. Returns: The resulting arrays. r-c.g|]}jj|Sr)r0r1)rrr2s rr z,build_arrays_from_tuples..s$ - - -1NBINA  - - -r!)r.r/)rbr2s @rbuild_arrays_from_tuplesres/  W % %B - - - - - - --r!r?)NN).....)......)rrrCNFF)__doc__r0typingrrrrrrr r r.opt_einsum.parserr opt_einsum.typingr r r _valid_chars_sizesdictzipr"strintr*r5boolrArerr!rrqs&& MMMMMMMMMMMMMMMMMMMM ((((((BBBBBBBBBB$ B B BD\62233htCH~.FRWXgilXlRm<cg !)$sCx.!9RZ[^R_ 9D  / /  / /  /  /  / 3- / /dm / 3$sCx. () / / /  / '*            3-  en  3=     "hh hh h  h  h 3- hhh 5hS#X. /sH}1E EFhhhhV .8 .S . . . . . .r!