|
|
#!/bin/bash
|
|
|
# ===================================
|
|
|
# A股/港股/美股 智能分析系统 - 测试脚本
|
|
|
# ===================================
|
|
|
#
|
|
|
# 使用方法：
|
|
|
# ./test.sh [测试场景]
|
|
|
#
|
|
|
# 测试场景：
|
|
|
# market - 仅大盘复盘
|
|
|
# a-stock - A股个股分析（茅台、平安银行）
|
|
|
# etf - etf分析(卫星etf 563230)
|
|
|
# hk-stock - 港股分析（腾讯、阿里）
|
|
|
# us-stock - 美股分析（苹果、特斯拉）
|
|
|
# mixed - 混合市场分析
|
|
|
# single - 单股模式测试
|
|
|
# dry-run - 仅获取数据不分析
|
|
|
# full - 完整流程测试
|
|
|
# quick - 快速测试（单只股票）
|
|
|
# all - 运行所有测试
|
|
|
#
|
|
|
# 示例：
|
|
|
# ./test.sh market # 测试大盘复盘
|
|
|
# ./test.sh us-stock # 测试美股分析
|
|
|
# ./test.sh quick # 快速测试
|
|
|
#
|
|
|
|
|
|
set -e
|
|
|
|
|
|
# 颜色定义
|
|
|
RED='\033[0;31m'
|
|
|
GREEN='\033[0;32m'
|
|
|
YELLOW='\033[1;33m'
|
|
|
BLUE='\033[0;34m'
|
|
|
NC='\033[0m' # No Color
|
|
|
|
|
|
# 打印带颜色的信息
|
|
|
info() {
|
|
|
echo -e "${BLUE}[INFO]${NC} $1"
|
|
|
}
|
|
|
|
|
|
success() {
|
|
|
echo -e "${GREEN}[SUCCESS]${NC} $1"
|
|
|
}
|
|
|
|
|
|
warn() {
|
|
|
echo -e "${YELLOW}[WARN]${NC} $1"
|
|
|
}
|
|
|
|
|
|
error() {
|
|
|
echo -e "${RED}[ERROR]${NC} $1"
|
|
|
}
|
|
|
|
|
|
header() {
|
|
|
echo ""
|
|
|
echo "=============================================="
|
|
|
echo -e "${GREEN}$1${NC}"
|
|
|
echo "=============================================="
|
|
|
echo ""
|
|
|
}
|
|
|
|
|
|
# 检查Python环境
|
|
|
check_python() {
|
|
|
if ! command -v python3 &> /dev/null; then
|
|
|
error "Python3 未安装"
|
|
|
exit 1
|
|
|
fi
|
|
|
info "Python版本: $(python3 --version)"
|
|
|
}
|
|
|
|
|
|
# 检查依赖
|
|
|
check_deps() {
|
|
|
info "检查依赖..."
|
|
|
python3 -c "import yfinance" 2>/dev/null || { warn "yfinance 未安装，美股测试可能失败"; }
|
|
|
python3 -c "import akshare" 2>/dev/null || { warn "akshare 未安装，A股/港股测试可能失败"; }
|
|
|
success "依赖检查完成"
|
|
|
}
|
|
|
|
|
|
# ==================== 测试场景 ====================
|
|
|
|
|
|
# 测试1: 大盘复盘
|
|
|
test_market() {
|
|
|
header "测试场景: 大盘复盘"
|
|
|
info "运行大盘复盘分析..."
|
|
|
python3 main.py --market-review "$@"
|
|
|
success "大盘复盘测试完成"
|
|
|
}
|
|
|
|
|
|
# 测试2: A股分析
|
|
|
test_a_stock() {
|
|
|
header "测试场景: A股分析"
|
|
|
info "分析A股: 600519(茅台), 000001(平安银行)"
|
|
|
python3 main.py --stocks 600519,000001 --no-market-review "$@"
|
|
|
success "A股分析测试完成"
|
|
|
}
|
|
|
|
|
|
# 测试2.5: ETF分析
|
|
|
test_etf() {
|
|
|
header "测试场景: ETF分析"
|
|
|
info "分析ETF: 563230(卫星ETF)"
|
|
|
python3 main.py --stocks 563230,512400 --no-market-review "$@"
|
|
|
success "ETF分析测试完成"
|
|
|
}
|
|
|
|
|
|
# 测试3: 港股分析
|
|
|
test_hk_stock() {
|
|
|
header "测试场景: 港股分析"
|
|
|
info "分析港股: hk00700(腾讯), hk09988(阿里)"
|
|
|
python3 main.py --stocks hk00700,hk09988 --no-market-review "$@"
|
|
|
success "港股分析测试完成"
|
|
|
}
|
|
|
|
|
|
# 测试4: 美股分析
|
|
|
test_us_stock() {
|
|
|
header "测试场景: 美股分析"
|
|
|
info "分析美股: AAPL(苹果), TSLA(特斯拉)"
|
|
|
# 允许透传参数，默认不带 --no-notify
|
|
|
python3 main.py --stocks AAPL --no-market-review "$@"
|
|
|
success "美股分析测试完成"
|
|
|
}
|
|
|
|
|
|
# 测试5: 混合市场
|
|
|
test_mixed() {
|
|
|
header "测试场景: 混合市场分析"
|
|
|
info "分析混合市场: 600519(A股), hk00700(港股), AAPL(美股)"
|
|
|
python3 main.py --stocks 600519,hk00700,AAPL --no-market-review
|
|
|
success "混合市场测试完成"
|
|
|
}
|
|
|
|
|
|
# 测试6: 单股推送模式
|
|
|
test_single() {
|
|
|
header "测试场景: 单股推送模式"
|
|
|
info "测试单股推送模式..."
|
|
|
python3 main.py --stocks 600519 --single-notify --no-market-review
|
|
|
success "单股推送模式测试完成"
|
|
|
}
|
|
|
|
|
|
# 测试7: dry-run模式
|
|
|
test_dry_run() {
|
|
|
header "测试场景: Dry-Run 模式"
|
|
|
info "仅获取数据，不进行AI分析..."
|
|
|
python3 main.py --stocks 600519,AAPL --dry-run --no-notify
|
|
|
success "Dry-Run 测试完成"
|
|
|
}
|
|
|
|
|
|
# 测试8: 完整流程
|
|
|
test_full() {
|
|
|
header "测试场景: 完整流程"
|
|
|
info "运行完整分析流程（个股+大盘）..."
|
|
|
python3 main.py --stocks 600519 --no-notify
|
|
|
success "完整流程测试完成"
|
|
|
}
|
|
|
|
|
|
# 测试9: 快速测试
|
|
|
test_quick() {
|
|
|
header "测试场景: 快速测试"
|
|
|
info "单只股票快速测试..."
|
|
|
python3 main.py --stocks 600519 --no-market-review
|
|
|
success "快速测试完成"
|
|
|
}
|
|
|
|
|
|
# 测试10: 代码识别测试
|
|
|
test_code_recognition() {
|
|
|
header "测试场景: 代码识别"
|
|
|
info "测试股票代码识别逻辑..."
|
|
|
|
|
|
python3 << 'PYTEST'
|
|
|
import sys
|
|
|
sys.path.insert(0, '.')
|
|
|
from data_provider.akshare_fetcher import _is_hk_code, _is_us_code
|
|
|
|
|
|
test_cases = [
|
|
|
# (代码, 预期HK, 预期US, 描述)
|
|
|
("AAPL", False, True, "美股-苹果"),
|
|
|
("TSLA", False, True, "美股-特斯拉"),
|
|
|
("BRK.B", False, True, "美股-伯克希尔B"),
|
|
|
("hk00700", True, False, "港股-腾讯"),
|
|
|
("HK09988", True, False, "港股-阿里"),
|
|
|
("600519", False, False, "A股-茅台"),
|
|
|
("000001", False, False, "A股-平安"),
|
|
|
]
|
|
|
|
|
|
print("\n股票代码识别测试:")
|
|
|
print("-" * 60)
|
|
|
all_pass = True
|
|
|
for code, exp_hk, exp_us, desc in test_cases:
|
|
|
is_hk = _is_hk_code(code)
|
|
|
is_us = _is_us_code(code)
|
|
|
hk_ok = is_hk == exp_hk
|
|
|
us_ok = is_us == exp_us
|
|
|
status = "✅" if (hk_ok and us_ok) else "❌"
|
|
|
all_pass = all_pass and hk_ok and us_ok
|
|
|
print(f"{status} {code:10} | HK:{is_hk:5} US:{is_us:5} | {desc}")
|
|
|
|
|
|
print("-" * 60)
|
|
|
print(f"{'✅ 所有测试通过!' if all_pass else '❌ 有测试失败!'}")
|
|
|
sys.exit(0 if all_pass else 1)
|
|
|
PYTEST
|
|
|
|
|
|
success "代码识别测试完成"
|
|
|
}
|
|
|
|
|
|
# 测试11: YFinance代码转换测试
|
|
|
test_yfinance_convert() {
|
|
|
header "测试场景: YFinance 代码转换"
|
|
|
info "测试YFinance代码转换逻辑..."
|
|
|
|
|
|
python3 << 'PYTEST'
|
|
|
import sys
|
|
|
sys.path.insert(0, '.')
|
|
|
from data_provider.yfinance_fetcher import YfinanceFetcher
|
|
|
|
|
|
fetcher = YfinanceFetcher()
|
|
|
|
|
|
test_cases = [
|
|
|
("AAPL", "AAPL", "美股"),
|
|
|
("tsla", "TSLA", "美股小写"),
|
|
|
("BRK.B", "BRK.B", "美股特殊"),
|
|
|
("hk00700", "0700.HK", "港股"),
|
|
|
("HK09988", "9988.HK", "港股大写"),
|
|
|
("600519", "600519.SS", "A股沪市"),
|
|
|
("000001", "000001.SZ", "A股深市"),
|
|
|
("300750", "300750.SZ", "A股创业板"),
|
|
|
]
|
|
|
|
|
|
print("\nYFinance 代码转换测试:")
|
|
|
print("-" * 60)
|
|
|
all_pass = True
|
|
|
for input_code, expected, desc in test_cases:
|
|
|
result = fetcher._convert_stock_code(input_code)
|
|
|
status = "✅" if result == expected else "❌"
|
|
|
all_pass = all_pass and (result == expected)
|
|
|
print(f"{status} {input_code:10} -> {result:12} (期望: {expected:12}) | {desc}")
|
|
|
|
|
|
print("-" * 60)
|
|
|
print(f"{'✅ 所有测试通过!' if all_pass else '❌ 有测试失败!'}")
|
|
|
sys.exit(0 if all_pass else 1)
|
|
|
PYTEST
|
|
|
|
|
|
success "YFinance 代码转换测试完成"
|
|
|
}
|
|
|
|
|
|
# 测试12: 语法检查
|
|
|
test_syntax() {
|
|
|
header "测试场景: Python 语法检查"
|
|
|
info "检查所有Python文件语法..."
|
|
|
|
|
|
python3 -m py_compile main.py src/config.py src/notification.py \
|
|
|
data_provider/akshare_fetcher.py \
|
|
|
data_provider/yfinance_fetcher.py \
|
|
|
bot/commands/analyze.py
|
|
|
|
|
|
success "语法检查通过"
|
|
|
}
|
|
|
|
|
|
# 测试13: Flake8 静态检查
|
|
|
test_flake8() {
|
|
|
header "测试场景: Flake8 静态检查"
|
|
|
info "运行 Flake8 检查严重错误..."
|
|
|
|
|
|
if command -v flake8 &> /dev/null; then
|
|
|
flake8 main.py src/config.py src/notification.py --select=F821,E999 --max-line-length=120
|
|
|
success "Flake8 检查通过"
|
|
|
else
|
|
|
warn "Flake8 未安装，跳过检查"
|
|
|
fi
|
|
|
}
|
|
|
|
|
|
# 运行所有测试
|
|
|
test_all() {
|
|
|
header "运行所有测试"
|
|
|
|
|
|
test_syntax
|
|
|
test_code_recognition
|
|
|
test_yfinance_convert
|
|
|
test_flake8
|
|
|
|
|
|
echo ""
|
|
|
info "以下测试需要网络和API配置，可能会失败:"
|
|
|
echo ""
|
|
|
|
|
|
test_dry_run || warn "Dry-Run 测试失败（可能是网络问题）"
|
|
|
test_quick || warn "快速测试失败（可能是API问题）"
|
|
|
|
|
|
success "所有测试完成!"
|
|
|
}
|
|
|
|
|
|
# ==================== 主程序 ====================
|
|
|
|
|
|
main() {
|
|
|
header "A股/港股/美股 智能分析系统 - 测试"
|
|
|
|
|
|
check_python
|
|
|
check_deps
|
|
|
|
|
|
case "${1:-help}" in
|
|
|
market)
|
|
|
shift
|
|
|
test_market "$@"
|
|
|
;;
|
|
|
a-stock|a_stock|astock)
|
|
|
shift
|
|
|
test_a_stock "$@"
|
|
|
;;
|
|
|
etf)
|
|
|
shift
|
|
|
test_etf "$@"
|
|
|
;;
|
|
|
hk-stock|hk_stock|hkstock|hk)
|
|
|
shift
|
|
|
test_hk_stock "$@"
|
|
|
;;
|
|
|
us-stock|us_stock|usstock|us)
|
|
|
shift
|
|
|
test_us_stock "$@"
|
|
|
;;
|
|
|
mixed|mix)
|
|
|
shift
|
|
|
test_mixed "$@"
|
|
|
;;
|
|
|
single)
|
|
|
shift
|
|
|
test_single "$@"
|
|
|
;;
|
|
|
dry-run|dryrun|dry)
|
|
|
shift
|
|
|
test_dry_run "$@"
|
|
|
;;
|
|
|
full)
|
|
|
shift
|
|
|
test_full "$@"
|
|
|
;;
|
|
|
quick|q)
|
|
|
shift
|
|
|
test_quick "$@"
|
|
|
;;
|
|
|
code|recognition)
|
|
|
shift
|
|
|
test_code_recognition "$@"
|
|
|
;;
|
|
|
yfinance|yf)
|
|
|
shift
|
|
|
test_yfinance_convert "$@"
|
|
|
;;
|
|
|
syntax)
|
|
|
shift
|
|
|
test_syntax "$@"
|
|
|
;;
|
|
|
flake8|lint)
|
|
|
shift
|
|
|
test_flake8 "$@"
|
|
|
;;
|
|
|
all)
|
|
|
shift
|
|
|
test_all "$@"
|
|
|
;;
|
|
|
help|--help|-h|*)
|
|
|
echo "使用方法: $0 [测试场景]"
|
|
|
echo ""
|
|
|
echo "测试场景:"
|
|
|
echo " market - 仅大盘复盘"
|
|
|
echo " a-stock - A股个股分析"
|
|
|
echo " etf - ETF分析"
|
|
|
echo " hk-stock - 港股分析"
|
|
|
echo " us-stock - 美股分析"
|
|
|
echo " mixed - 混合市场分析"
|
|
|
echo " single - 单股推送模式"
|
|
|
echo " dry-run - 仅获取数据"
|
|
|
echo " full - 完整流程"
|
|
|
echo " quick - 快速测试（推荐）"
|
|
|
echo " code - 代码识别测试"
|
|
|
echo " yfinance - YFinance转换测试"
|
|
|
echo " syntax - 语法检查"
|
|
|
echo " flake8 - 静态检查"
|
|
|
echo " all - 运行所有测试"
|
|
|
echo ""
|
|
|
echo "示例:"
|
|
|
echo " $0 quick # 快速测试"
|
|
|
echo " $0 us-stock # 测试美股"
|
|
|
echo " $0 code # 测试代码识别"
|
|
|
echo " $0 all # 运行所有测试"
|
|
|
;;
|
|
|
esac
|
|
|
}
|
|
|
|
|
|
main "$@"
|
