__author__ = "Jeroen Van Der Donckt"

import pandas as pd
import pytest
from sklearn.datasets import make_classification, make_regression


@pytest.fixture
def dummy_classification() -> pd.DataFrame:
    X, y = make_classification(
        n_samples=500,
        n_features=10,
        n_informative=2,
        n_repeated=0,
        n_redundant=0,
        random_state=42,
        shuffle=False,
    )
    X = pd.DataFrame(
        X, columns=[f"informative_{i}" for i in range(2)] + [f"random_{i}" for i in range(8)]
    )
    return X, y


@pytest.fixture
def dummy_regression() -> pd.DataFrame:
    X, y = make_regression(
        n_samples=500, n_features=10, n_informative=2, random_state=42, shuffle=False
    )
    X = pd.DataFrame(
        X, columns=[f"informative_{i}" for i in range(2)] + [f"random_{i}" for i in range(8)]
    )
    return X, y
