34d09d9d34
Some checks failed
Deploy to Production / deploy (push) Failing after 6m46s
- KOSPIMarketStateDetector: KOSPI MA 기반 시장 상태 판단 (bull/neutral/bear/crash) - VolumeScreener: 거래대금 2000억+ 스크리닝 (상한가 우선, 희소성 체크, 대형주 예외) - SectorPortfolioManager: 섹터 기반 비중 배분 - KJBScreeningSignalGenerator: 눌림목 진입, 5MA 손절, 단계적 익절 - KISTradeExecutor: KIS API 자동 매수/매도 (기본값 모의투자) - ScreeningSignal / AutoOrder DB 모델 추가 - screening API 엔드포인트 추가 - 스케줄러 잡 3종 추가 (08:30/5분/15:35) - Price.trading_value 컬럼 추가 - MarketIndex 테이블 추가 (KOSPI/KOSDAQ 지수 일봉) - IndexCollector 추가 (일일 수집 잡 등록) - intraday_exit_check 시간 필터 추가 (09:05~15:20 KST) - 드라이런 스크립트 추가 (scripts/screening_dryrun.py)
27 lines
758 B
Python
27 lines
758 B
Python
"""Unit tests for IndexCollector."""
|
|
from unittest.mock import patch
|
|
|
|
import pandas as pd
|
|
|
|
from app.services.collectors.index_collector import IndexCollector
|
|
|
|
|
|
def make_mock_df():
|
|
data = {
|
|
"시가": [2500.0],
|
|
"고가": [2550.0],
|
|
"저가": [2480.0],
|
|
"종가": [2520.0],
|
|
"거래량": [500000000],
|
|
"거래대금": [5000000000000],
|
|
}
|
|
df = pd.DataFrame(data, index=pd.DatetimeIndex([pd.Timestamp("2024-01-02")], name="날짜"))
|
|
return df
|
|
|
|
|
|
def test_index_collector_collect(db):
|
|
with patch("pykrx.stock.get_index_ohlcv", return_value=make_mock_df()):
|
|
collector = IndexCollector(db, start_date="20240102", end_date="20240102")
|
|
count = collector.collect()
|
|
assert count >= 1
|