51 lines
1.6 KiB
Python
51 lines
1.6 KiB
Python
import os
|
||
import sys
|
||
|
||
import pandas as pd
|
||
|
||
sys.path.insert(0, os.path.dirname(os.path.dirname(__file__)))
|
||
|
||
from app.services.altcoin_screener import detect_volume_price_fly
|
||
from app.services.altcoin_confirm import detect_volume_price_fly_1h
|
||
|
||
|
||
def _sample_df(stale_age_hours=9):
|
||
rows = []
|
||
base_time = pd.Timestamp('2026-05-10 00:00:00')
|
||
# 12根 recent + 前20均量所需历史；默认低量小阳/小阴
|
||
for i in range(32):
|
||
rows.append({
|
||
'time': base_time + pd.Timedelta(hours=i),
|
||
'open': 1.0,
|
||
'high': 1.01,
|
||
'low': 0.99,
|
||
'close': 1.002,
|
||
'volume': 100.0,
|
||
})
|
||
# recent 中距离最新 stale_age_hours 的那根制造放量大阳线
|
||
target_idx = len(rows) - 1 - stale_age_hours
|
||
rows[target_idx].update({'open': 1.0, 'high': 1.3, 'low': 0.99, 'close': 1.25, 'volume': 2000.0})
|
||
return pd.DataFrame(rows)
|
||
|
||
|
||
def test_stale_1h_volume_price_fly_not_counted_as_current_signal():
|
||
df = _sample_df(stale_age_hours=9)
|
||
result = detect_volume_price_fly(df)
|
||
assert result['vp_fly_count'] == 0
|
||
assert result['stale_vp_fly_count'] >= 1
|
||
assert result['stale_vp_fly_details'][0]['age_hours'] == 9
|
||
|
||
|
||
def test_confirm_layer_stale_volume_price_fly_not_confirmed():
|
||
df = _sample_df(stale_age_hours=9)
|
||
result = detect_volume_price_fly_1h(df)
|
||
assert result['vp_fly_count'] == 0
|
||
assert result['stale_vp_fly_count'] >= 1
|
||
|
||
|
||
def test_recent_1h_volume_price_fly_is_counted():
|
||
df = _sample_df(stale_age_hours=1)
|
||
result = detect_volume_price_fly(df)
|
||
assert result['vp_fly_count'] == 1
|
||
assert result['vp_fly_details'][0]['age_hours'] == 1
|