import pytest
import numpy as np
from src.data import CreditCardDataProcessor, load_data
from pathlib import Path


def test_data_processor_initialization():
    processor = CreditCardDataProcessor("data/creditcard.csv")
    assert processor.file_path == "data/creditcard.csv"
    assert processor.data is None


def test_load_data():
    processor = CreditCardDataProcessor("data/creditcard.csv")
    processor.load_data()
    assert processor.data is not None
    assert processor.data.height > 0
    assert "Class" in processor.data.columns


def test_validate_data():
    processor = CreditCardDataProcessor("data/creditcard.csv")
    processor.load_data()
    processor.validate_data()
    assert processor.data is not None


def test_split_data_by_time():
    processor = CreditCardDataProcessor("data/creditcard.csv")
    processor.load_data()
    train_data, test_data = processor.split_data_by_time(test_ratio=0.2)
    assert train_data is not None
    assert test_data is not None
    assert train_data.height > test_data.height


def test_prepare_features_labels():
    processor = CreditCardDataProcessor("data/creditcard.csv")
    processor.load_data()
    processor.split_data_by_time()
    processor.prepare_features_labels()
    assert processor.train_features is not None
    assert processor.train_labels is not None
    assert processor.test_features is not None
    assert processor.test_labels is not None


def test_load_data_function():
    processor = load_data("data/creditcard.csv")
    assert processor.data is not None
    assert processor.train_features is not None
    assert processor.test_features is not None


def test_get_statistics():
    processor = load_data("data/creditcard.csv")
    stats = processor.get_statistics()
    assert "总记录数" in stats
    assert "特征数" in stats
    assert "欺诈交易数" in stats
    assert "非欺诈交易数" in stats
    assert stats["总记录数"] > 0