^jk UdZddlmZmZmZmZddlZddlm Z m Z ddl m Z edZ eed<d dd d e d ed edededdf dZde de defdZddeededefdZddeededefdZy)zHelpers for test code.)AnyLiteralSequence TypeAliasN)_is_cupy_alike import_cupy)DMatrix)cpucudaDevicegHz>)rtolatoldeviceabrrreturnc|dk(r:t|s/t|s$tjj||||yt }|jj||||y)z)Dispatch the assert_allclose for devices.r )rrN)rnptestingassert_allcloser )rrrrrcps `/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/xgboost/testing/utils.pyrr sV~a09J ""1ad"> ] ""1ad">lhsrhscN|j}|j}ttj|j|jtj|j |j tj|j |j fS)z9Assert whether two DMatrices contain the same predictors.)get_dataallr array_equaldataindicesindptr)rrlcsrrcsrs rpredictor_equalr&sl <<>D <<>D  NN499dii 0 NN4<< 6 NN4;; 4  rL tolerancec Ftfdt||ddDS)z*Values in the sequence are not increasing.c34K|]\}}||z kywN.0xyr(s r z!non_increasing..'s=tq!A"=sNrzipr'r(s `rnon_increasingr6%s =s1ae}= ==rc Ftfdt||ddDS)z*Values in the sequence are not decreasing.c36K|]\}}||z  k\ywr+r,r-s rr1z!non_decreasing..,s ?AA9*$?sr2Nr3r5s `rnon_decreasingr9*s ?Q!"? ??r)g-C6?)__doc__typingrrrrnumpyrcompatrr corer r __annotations__floatrboolr&r6r9r,rrrBs440M* *6:? ??"?-2?AF? ?  w 4 >huo>%>4> @huo@%@4@r