import pandas as pd
import numpy as np

# 设置随机种子
np.random.seed(42)

# 生成 200 天的时间序列
dates = pd.date_range(start='2023-01-01', periods=200, freq='D')

# 构造数据
trend = np.linspace(0, 50, 200)
seasonality = 10 * np.sin(np.linspace(0, 3.14 * 2 * (200/7), 200))
noise = np.random.normal(0, 2, 200)

sales = 100 + trend + seasonality + noise
ad_cost = sales * 0.5 + np.random.normal(0, 5, 200)
temperature = 30 - trend * 0.2 + np.random.normal(0, 3, 200)

# 创建 DataFrame
df = pd.DataFrame({
    'date': dates,
    'sales': sales,
    'ad_cost': ad_cost,
    'temperature': temperature
})

# 保存
df.to_csv('complex_test.csv', index=False)
print("✅ 成功生成测试文件: complex_test.csv")