9j*%SSKJr SSKrSSKrSSKrSSKrSSKrSSKJ r SSKJ r SSKJ r SSK r SSK r SSKJr SSKJr \"5(a SSKr\R&R(rS\S 'O \R.rS"S jrS#S jrS$S jr\ \ R6S 4Sjr\ \ R64Sjr\ \ R6SSS 4SjrS\ R6SSS4Sjr\ \ R64Sjr \ \ R64SS.Sjjr!\RDS5r#"SS\RH5r%"SS\%5r&"SS\%5r'\RDSS .S!j5r(\(r)g)%) annotationsN)Callable)PackageNotFoundError)mock)internal) is_availablez-Callable[..., Callable[[Callable], Callable]]_skipifcTSSRU53n[[U6(+US9$)aRun a test case only when given requirements are satisfied. .. admonition:: Example This test case runs only when `numpy>=1.18` is installed. >>> from cupy import testing ... ... ... class Test(unittest.TestCase): ... @testing.with_requires("numpy>=1.18") ... def test_for_numpy_1_18(self): ... pass Args: requirements: A list of string representing requirement condition to run a given test case. z requires: ,)reason)joinr installed) requirementsmsgs N/home/wildlama/miniconda3/lib/python3.13/site-packages/cupy/testing/_helper.py with_requiresrs/( sxx -. /C y,// <<cSSKJn UHanU"U5n[RR UR 5nURnU(dMJURUSS9(aMa g g![ a  gf=f)zReturns True if the current environment satisfies the specified package requirement. Args: specifiers: Version specifiers (e.g., `numpy>=1.20.0`). r) RequirementFT) prereleases) packaging.requirementsr importlibmetadataversionnamer specifiercontains) specifiersrspecreqfoundexpecteds rrr3s{3$ &&..sxx8E== 8X..u$.GG  $  s)A// A=<A=c[SU35$)zReturns True if numpy version satisfies the specified criteria. Args: version_range: A version specifier (e.g., `>=1.13.0`). numpy)r) version_ranges rnumpy_satisfiesr&Js u]O, --rCc4[R"U5n[R"S[R"U5S-S5nUS:Xa US-S:HnOUR S:XaXDS--nUR URU5RU5US9$)aBReturns an array with given shape, array module, and dtype. Args: shape(tuple of int): Shape of returned ndarray. xp(numpy or cupy): Array module to use. dtype(dtype): Dtype of returned ndarray. order({'C', 'F'}): Order of returned ndarray. Returns: numpy.ndarray or cupy.ndarray: The array filled with :math:`1, \cdots, N` with specified dtype with given shape, array module. Here, :math:`N` is the size of the returned array. If ``dtype`` is ``numpy.bool_``, evens (resp. odds) are converted to ``True`` (resp. ``False``). ?rc?)order) r$dtypearangerprodkindarrayastypereshape)shapexpr/r.as r shaped_aranger9Ss$ KK E Q e,q0!4A | EQJ s  BJ 88AHHUO++E2%8 @@rc4[R"U5n[R"U5n[R"USS5nUS:Xa US-S:HnOUR S:XaXDS--nUR URU5RU55$)aReturns an array filled with decreasing numbers. Args: shape(tuple of int): Shape of returned ndarray. xp(numpy or cupy): Array module to use. dtype(dtype): Dtype of returned ndarray. Returns: numpy.ndarray or cupy.ndarray: The array filled with :math:`N, \cdots, 1` with specified dtype with given shape, array module. Here, :math:`N` is the size of the returned array. If ``dtype`` is ``numpy.bool_``, evens (resp. odds) are converted to ``True`` (resp. ``False``). rr*r+r,r-) r$r/rr1r0r2r3r4r5)r6r7r/sizer8s rshaped_reverse_aranger=ns KK E == D T1b!A | EQJ s  BJ 88AHHUO++E2 33r cL[RRU5n[R"U5nUS:XaUR SUS9nOIUR S:Xa'UR "U6SUR "U6--nXs-nOUR "U6U-nURXrUS9$)ayReturns an array filled with random values. Args: shape(tuple): Shape of returned ndarray. xp(numpy or cupy): Array module to use. dtype(dtype): Dtype of returned ndarray. scale(float): Scaling factor of elements. seed(int): Random seed. Returns: numpy.ndarray or cupy.ndarray: The array with given shape, array module, If ``dtype`` is ``numpy.bool_``, the elements are independently drawn from ``True`` and ``False`` with same probabilities. Otherwise, the array is filled with samples independently and identically drawn from uniform distribution over :math:`[0, scale)` with specified dtype. r*r+)r<r,r-)r/r.)r$random RandomStater/randintr2randasarray)r6r7r/scaleseedr.rngr8s r shaped_randomrHs. ,, " "4 (C KK E | KKK & s  HHe rCHHe$44 4  HHe u $ ::aE: 22rg{Gz?coocSSKnSSKnUcURRnUupURR XX5S9R U5n UR U 5RU5$![a [SUS35ef=f)aReturns an array filled with random values. Args: shape (tuple): Shape of returned sparse matrix. sp (scipy.sparse or cupyx.scipy.sparse): Sparse matrix module to use. dtype (dtype): Dtype of returned sparse matrix. density (float): Density of returned sparse matrix. format (str): Format of returned sparse matrix. seed (int): Random seed. Returns: The sparse matrix with given shape, array module, rN) random_statezModule z' does not have the expected sparse APIs) scipy.sparsecupyx.scipy.sparsescipysparser@r4 coo_matrixasformatAttributeError ValueError) r6spr/densityformatrFrNcupyxn_rowsn_colsr8s rshaped_sparse_randomrZs  z [[  NF   44:F5MP}}Q((00 P72$&MNOOPs A00B c\[R"U5n[R"U5nUS:Xa[US5n[ US5nOUR S:Xa [US5n[R "XU5nURURU5RU55$)aDReturns an array with given shape, array module, and dtype. Args: start (int): The starting value. stop (int): The end value. shape (tuple of int): Shape of returned ndarray. xp (numpy or cupy): Array module to use. dtype (dtype): Dtype of returned ndarray. Returns: numpy.ndarray or cupy.ndarray: r*rr)u) r$r/r1maxminr2linspacer3r4r5)startstopr6r7r/r<r8s rshaped_linspacerbs KK E ::e D |E1 4| s E1  uD)A 88AHHUO++E2 33r)singular_valuesc[U5S::a[SRU55eUc [S5eUR U5n[ R "U5nURS;a[SRU55eURU5(d [S5eUS:R5(a [S5eURR"U6nURS :XaUS URR"U6--nURRUS S 9upVnURX6R5nUR!S XXU5nUR#U5$)aqReturns a matrix with specified singular values. Generates a random matrix with given singular values. This function generates a random NumPy matrix (or a stack of matrices) that has specified singular values. It can be used to generate the inputs for a test that can be instable when the input value behaves bad. Notation: denote the shape of the generated array by :math:`(B..., M, N)`, and :math:`K = min\{M, N\}`. :math:`B...` may be an empty sequence. Args: shape (tuple of int): Shape of the generated array, i.e., :math:`(B..., M, N)`. xp (numpy or cupy): Array module to use. dtype: Dtype of the generated array. singular_values (array-like): Singular values of the generated matrices. It must be broadcastable to shape :math:`(B..., K)`. Returns: numpy.ndarray or cupy.ndarray: A random matrix that has specific singular values. r)z.shape {} is invalid for matrices: too few axeszsingular_values is not givenfczdtype {} is not supportedzsingular_values is not realrz negative singular value is givenr,r-F) full_matricesz...ik,...k,...kj->...ij)lenrSrV TypeErrorrDr$r/r2 isrealobjanyr@randnlinalgsvd broadcast_tor6einsumr4) r6r7r/rcr8r\svhsvs rgenerate_matrixrssA0 5zQ < C CE J  677jj1O KK E zz3::5ABB << ( (566!  "";<< A zzS RYY__e,, ,yy}}Qe}4HA" '' 2B +QB7A 88E?rc#>^# [R"SS9n[R"S5 Sv SSS5 [U4SjW55(agTRn[SU-5e!,(df  ND=f![ a [ T5nN6f=f7f)NT)recordalwaysc3P># UHn[URT5v M g7fN) isinstancemessage).0mr"s r assert_warns..s 6Aq:aii * *As#&z%s not triggerred)warningscatch_warnings simplefilterrj__name__rRstrAssertionError)r"wexc_names` r assert_warnsrs   -h'  . 6A 666!$$ ,x7 88 . - !x=!s?BA1#B B#B1 A?;BBBBBcD\rSrSr\S5r\S5r\S5rSrg)NumpyAliasTestBasei)c[5erx)NotImplementedErrorselfs rfuncNumpyAliasTestBase.func+s !##rc6[[UR5$rx)getattrcupyrrs r cupy_funcNumpyAliasTestBase.cupy_func/stTYY''rc6[[UR5$rx)rr$rrs r numpy_funcNumpyAliasTestBase.numpy_func3sudii((rN) r __module__ __qualname____firstlineno__propertyrrr__static_attributes__rrrrr)s< $$(())rrc \rSrSrSrSrSrg)NumpyAliasBasicTestBasei8ct[RnU"UR5U"UR5:Xdegrx)inspect signaturerr)rfs r test_argspec$NumpyAliasBasicTestBase.test_argspec:s-    Adoo$6666rcURnURn[US5(deURceURS:wdeURURLdeg)N__doc__)rrhasattrr)rrrs rtest_docstring&NumpyAliasBasicTestBase.test_docstring>sgNN __ y),,,,  ,,,  B&&&  (:(::::rrN)rrrrrrrrrrrr8s 7;rrc\rSrSrSrSrg)NumpyAliasValuesTestBaseiGcpUR"UR6UR"UR6:Xdegrx)rargsrrs r test_values$NumpyAliasValuesTestBase.test_valuesIs*~~tyy)T__dii-HHHHrrN)rrrrrrrrrrrGsIrrr)) times_calledc/# [R"U0UD6nSv URU:XdeSSS5 g!,(df  g=f7f)aA handy wrapper for unittest.mock to check if a function is called. Args: *args: Arguments of `mock.patch`. times_called (int): The number of times the function should be called. Default is ``1``. **kwargs: Keyword arguments of `mock.patch`. N)rpatch call_count)rrkwargshandles rassert_function_is_calledrMs< T $V $   L000 % $ $sA : A  AA )rrreturnzCallable[[Callable], Callable])rrrbool)r%rrr)* __future__r contextlibimportlib.metadatarrunittestrcollections.abcrrrr$r cupy._corercupy.testing._pytest_implrpytestmarkskipifr __annotations__skipIfrrr&float32r9r=rHrZrbrscontextmanagerrTestCaserrrrAssertFunctionIsCalledrrrrsM"$3 2>>=C[[=O=OG :OooG=0..! SA6%) 46emm2AS 3HemmUP>,0u}}42emm3@D3l  9 9 )** ) ;0 ;I1I  23 1 1 3r