
    =[g                     4    d Z ddlZddlZddlmZ deddfdZy)z*Tests related to the `DataIter` interface.    N)testingdevicereturnc                    t        j                  ddd      \  }}}| j                  d      rt        j                  dddd      \  }}nt        j                  dddd      \  }}t        j                  ddd	d      \  }}|||g}|||g}	| j                  d      r"d
dl}
|D cg c]  }|
j                  |       }}t        j                  ||	dd      }t        j                  |      }t        j                  dddd      \  }}|||g}|||g}	t        j                  |d
      }t        j                  |	d
      }t        j                  ||      }t        j                  ||      sJ yc c}w )zCheck QDM with mixed batches.      Fcuda   g?Ti   g?r   N)axis)tmmake_regression
startswithmake_sparse_regressioncupyarrayIteratorForTestxgboostQuantileDMatrixnpconcatenatepredictor_equal)r   X_0y_0_X_1y_1X_2y_2XycpbatchitXy_0X_arry_arrXy_1s                    T/var/www/html/bid-api/venv/lib/python3.12/site-packages/xgboost/testing/data_iter.pyrun_mixed_sparsityr)   	   s^   $$S"e4KCa ,,S"c4@S,,S"c5AS((b#t<HC	c3A	c3A *+,!RXXe_!,			Aq$	-B""2&D((b#t<HC	c3A	c3ANN11%ENN11%E""5%0DdD))) -s   "E+)__doc__numpyr   r   r   r   strr)        r(   <module>r/      s$    0   !*s *t *r.   