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)