#!/bin/bash

# Linux/Mac脚本 - 运行缺失值处理功能测试

echo "========================================"
echo "缺失值处理功能 - 自动化测试"
echo "========================================"
echo

# 检查Python是否安装
if ! command -v python3 &> /dev/null; then
    echo "[错误] Python未安装"
    exit 1
fi

echo "[1/3] 检查Python服务状态..."
if ! curl -s http://localhost:8001/health > /dev/null 2>&1; then
    echo "[警告] Python服务未运行，请先启动服务："
    echo "   cd extraction_service"
    echo "   python main.py"
    echo
    exit 1
fi
echo "[OK] Python服务运行正常"
echo

echo "[2/3] 检查依赖..."
python3 -c "import pandas, numpy, requests" 2> /dev/null
if [ $? -ne 0 ]; then
    echo "[警告] 缺少依赖，正在安装..."
    pip3 install pandas numpy requests
fi
echo "[OK] 依赖检查完成"
echo

echo "[3/3] 运行测试..."
echo
python3 test_fillna_operations.py

echo
echo "========================================"
echo "测试完成"
echo "========================================"