o X۷i@sddlmZddlZddlZddlZddlZ  ddddd Z dd d ZdddZdddZ ddddddZ dddZ dddddZ dS)) annotationsNHz>TF)strictc Csxtjtjdkrtjjt|t|||||||ddS|r't dt tjjt|t||||||ddS)a9Raises an AssertionError if objects are not equal up to desired tolerance. Args: actual(numpy.ndarray or cupy.ndarray): The actual object to check. desired(numpy.ndarray or cupy.ndarray): The desired, expected object. rtol(float): Relative tolerance. atol(float): Absolute tolerance. err_msg(str): The error message to be printed in case of failure. verbose(bool): If ``True``, the conflicting values are appended to the error message. .. seealso:: :func:`numpy.testing.assert_allclose` 2.0.0)rtolatol equal_nanerr_msgverboserN`cupy.testing.assert_allclose` does not support `strict` option with NumPy v1.)rrr r r N) numpylib NumpyVersion __version__testingassert_allclosecupyasnumpywarningswarnRuntimeWarning)actualdesiredrrr r r rrI/home/ubuntu/vllm_env/lib/python3.10/site-packages/cupy/testing/_array.pyr s   rcCs&tjjt|t||||ddS)a Raises an AssertionError if objects are not equal up to desired precision. Args: x(numpy.ndarray or cupy.ndarray): The actual object to check. y(numpy.ndarray or cupy.ndarray): The desired, expected object. decimal(int): Desired precision. err_msg(str): The error message to be printed in case of failure. verbose(bool): If ``True``, the conflicting values are appended to the error message. .. seealso:: :func:`numpy.testing.assert_array_almost_equal` )decimalr r N)r rassert_array_almost_equalrr)rrrr r rrrr3s rcCs"tjjt|t||ddS)afCompare two arrays relatively to their spacing. Args: x(numpy.ndarray or cupy.ndarray): The actual object to check. y(numpy.ndarray or cupy.ndarray): The desired, expected object. nulp(int): The maximum number of unit in the last place for tolerance. .. seealso:: :func:`numpy.testing.assert_array_almost_equal_nulp` )nulpN)r rassert_array_almost_equal_nulprr)xyr rrrr!Hs  r!cCs$tjjt|t|||ddS)aCheck that all items of arrays differ in at most N Units in the Last Place. Args: a(numpy.ndarray or cupy.ndarray): The actual object to check. b(numpy.ndarray or cupy.ndarray): The desired, expected object. maxulp(int): The maximum number of units in the last place that elements of ``a`` and ``b`` can differ. dtype(numpy.dtype): Data-type to convert ``a`` and ``b`` to if given. .. seealso:: :func:`numpy.testing.assert_array_max_ulp` )maxulpdtypeN)r rassert_array_max_ulprr)abr$r%rrrr&Vs  r&)r strides_checkcCstjtjdkrtjjt|t||||dn|r#t dt tjjt|t|||d|rd|j |j krfdg}|rI|dd|g}|r]| d |j | d |j td |d Sd S) aRaises an AssertionError if two array_like objects are not equal. Args: x(numpy.ndarray or cupy.ndarray): The actual object to check. y(numpy.ndarray or cupy.ndarray): The desired, expected object. strides_check(bool): If ``True``, consistency of strides is also checked. err_msg(str): The error message to be printed in case of failure. verbose(bool): If ``True``, the conflicting values are appended to the error message. strict(bool): If ``True``, raise an AssertionError when either the shape or the data type of the array_like objects does not match. Requires NumPy version 1.24 or above. .. seealso:: :func:`numpy.testing.assert_array_equal` z1.24.0r r rr r r zStrides are not equal:r z x: {}z y: {} N)r rrrrassert_array_equalrrrrrstridesappendformatAssertionErrorjoin)rrr r rr)msgrrrr.fs2 r.cCst|}t|}||urtd|||ttfvr!td|t|t|kr-tdt||D]\}}tjj t |t |||dq2dS)aCompares lists of arrays pairwise with ``assert_array_equal``. Args: x(array_like): Array of the actual objects. y(array_like): Array of the desired, expected objects. err_msg(str): The error message to be printed in case of failure. verbose(bool): If ``True``, the conflicting values are appended to the error message. Each element of ``x`` and ``y`` must be either :class:`numpy.ndarray` or :class:`cupy.ndarray`. ``x`` and ``y`` must have same length. Otherwise, this function raises ``AssertionError``. It compares elements of ``x`` and ``y`` pairwise with :func:`assert_array_equal` and raises error if at least one pair is not equal. .. seealso:: :func:`numpy.testing.assert_array_equal` zZMatching types of list or tuple are expected, but were different types (xlist:{} ylist:{})z%List or tuple is expected, but was {}zList size is differentr+N) typer2r1listtuplelenzipr rr.rr)xlistylistr r x_typey_typer"r#rrrassert_array_list_equals& r>cCsltjtjdkrtjjt|t||||ddS|r$t dt tjjt|t|||ddS)aRaises an AssertionError if array_like objects are not ordered by less than. Args: x(numpy.ndarray or cupy.ndarray): The smaller object to check. y(numpy.ndarray or cupy.ndarray): The larger object to compare. err_msg(str): The error message to be printed in case of failure. verbose(bool): If ``True``, the conflicting values are appended to the error message. .. seealso:: :func:`numpy.testing.assert_array_less` rr*r r+N) r rrrrassert_array_lessrrrrr)r"r#r r rrrrr?s   r?)rrTrT)rrT)r)rN)rT) __future__rrr numpy.testingrrrr!r&r.r>r?rrrrs(  '   0%