def test_buffer_dtype_mismatch_error():
with pytest.raises(ValueError, match='Buffer dtype mismatch'):
ArrayDataset64(X32, y32, sample_weight32, seed=42),
with pytest.raises(ValueError, match='Buffer dtype mismatch'):
ArrayDataset32(X64, y64, sample_weight64, seed=42),
with pytest.raises(ValueError, match='Buffer dtype mismatch'):
CSRDataset64(X_csr32.data, X_csr32.indptr, X_csr32.indices, y32,
sample_weight32, seed=42),
with pytest.raises(ValueError, match='Buffer dtype mismatch'):
CSRDataset32(X_csr64.data, X_csr64.indptr, X_csr64.indices, y64,
sample_weight64, seed=42),
def make_dense_dataset_64():
return ArrayDataset64(X64, y64, sample_weight64, seed=42)