import pytest
import numpy as np

@pytest.fixture
def classification_data():
    """生成二分类数据：200个样本，5个特征，最后一列是二分类标签（0或1）"""
    np.random.seed(42)
    X = np.random.randn(200, 5)
    # 根据特征和生成标签：如果第一列 > 0 则为 1，否则为 0
    y = (X[:, 0] > 0).astype(int)
    data = np.column_stack([X, y])
    return data