o װij@s^ddlZddlZ  dddZddd Zdd d Zdd dZdddZdddZdddZ dS)NHz>TcCs(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` )rtolatolerr_msgverboseN)numpytestingassert_allclosecupyasnumpy)actualdesiredrrrrrG/home/ubuntu/.local/lib/python3.10/site-packages/cupy/testing/_array.pyr s r cCs&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` )decimalrrN)rr assert_array_almost_equalr r )xyrrrrrrrs  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)rr assert_array_almost_equal_nulpr r )rrrrrrr/s  rcC$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)rr assert_array_max_ulpr r )abrrrrrr=s  rFcKstjjt|t|f||d||rE|j|jkrGdg}|r*|dd|g}|r>|d|j|d|jtd |dSdS) 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` rrzStrides are not equal:r z x: {}z y: {} N) rr assert_array_equalr r stridesappendformatAssertionErrorjoin)rrrr strides_checkkwargsmsgrrrr"Ms& 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 differentrN) typer&r%listtuplelenziprr r"r r )xlistylistrrx_typey_typerrrrrassert_array_list_equalns& r4cCr)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` rN)rr assert_array_lessr r )rrrrrrrr5s  r5)rrrT)rrT)r)rN)rTF)rT) numpy.testingrr r rrrr"r4r5rrrrs     !%