(
n_samples_per_batch: int,
n_features: int,
n_batches: int,
subsample: bool,
)
| 170 | ) |
| 171 | @settings(deadline=None, max_examples=10, print_blob=True) |
| 172 | def test_data_iterator( |
| 173 | n_samples_per_batch: int, |
| 174 | n_features: int, |
| 175 | n_batches: int, |
| 176 | subsample: bool, |
| 177 | ) -> None: |
| 178 | run_data_iterator( |
| 179 | n_samples_per_batch, |
| 180 | n_features, |
| 181 | n_batches, |
| 182 | "approx", |
| 183 | subsample, |
| 184 | "cpu", |
| 185 | False, |
| 186 | False, |
| 187 | ) |
| 188 | run_data_iterator( |
| 189 | n_samples_per_batch, |
| 190 | n_features, |
| 191 | n_batches, |
| 192 | "hist", |
| 193 | subsample, |
| 194 | "cpu", |
| 195 | False, |
| 196 | False, |
| 197 | ) |
| 198 | |
| 199 | |
| 200 | class IterForCacheTest(xgb.DataIter): |
nothing calls this directly
no test coverage detected