o 9jl @sdZddlZddlmmZddlZddlZddl m Z m Z m Z ej dej ej ej eddlZd ddZGdddZedkrUddlZeed gdSdS) ai test_momentum_classification.py Regression tests for the 2026-08-20 change that wires momentum_slope_class into compute_health()'s raw_state classification: 1. HEALTHY now requires momentum not be fading (in addition to the existing price/VWAP/volume/structure requirements). 2. A corroborated-only path reclassifies price_slope=positive + momentum-fading readings to STALE when at least one other weak signal agrees (volume declining, lower-highs/lower-lows, or price already lost VWAP) -- deliberately NOT on momentum alone, matching the module's existing require-multiple-signals philosophy. 3. Fading momentum with NO corroborating signal should NOT downgrade a reading -- single-signal noise should still fall through to WATCH, same as before this change. Run: python -m pytest tests/test_momentum_classification.py -v N)datetime timedeltatimezonec Csl|p tdddddtjd}g}tt||D]\}\}}||t|d|d|d |d ||d q|S) Ni )tzinfo)minutesg{Gz?g{Gz?)tohlcv)rrutc enumeratezipappendr)pricesvolumesstartbarsiprr^/var/www/screener/trade/premarket_backup_2026-09-08_2010/tests/test_momentum_classification.py make_barss  rc@s,eZdZddZddZddZddZd S) TestMomentumFadingClassificationc Csgd}gd}t||}tjd|dd}|j}d}||k}|sYtd|fd||fd tvs5t|r:t |nd t |t |d }t d d d |i} t t | d}}}|j d} | } | syddt | i} t t | d} } dS)zA clean, accelerating uptrend (increasing bar-over-bar gains, rising volume, no pullback) should still classify HEALTHY -- confirms the new momentum gate doesn't break the normal case.)皙 @gQ @g)\ @Q @ @fffff @gQ!@g)\(!@)i'*i2i> Na}@TEST順avg_vol_baselinenegative!=)z<%(py2)s {%(py2)s = %(py0)s.momentum_slope_class } != %(py5)sreadingpy0py2py5z>test setup should produce accelerating (non-negative) momentum >assert %(py7)spy7Nmomentum_fadingassert not %(py1)spy1)rintraday_healthcompute_healthmomentum_slope_class @pytest_ar_call_reprcompare @py_builtinslocals_should_repr_global_name _saferepr_format_assertmsgAssertionError_format_explanationflags) selfrrrr/ @py_assert1 @py_assert4 @py_assert3 @py_format6 @py_format8 @py_assert0 @py_assert2 @py_format3rrr)test_healthy_requires_non_fading_momentum+sH  @zJTestMomentumFadingClassification.test_healthy_requires_non_fading_momentumcCszgd}gd}t||}tjd|dd}|jdr|jd}| }|s4dd t|i}tt|d }}|jd }| }|sRdd t|i}tt|d }}|jd }|smd d t|i}tt|d }|j} d} | | k} | st d| fd| | fdt vst |rt|ndt| t| d} t ddd| i} tt| d } } } d Sd S)zFading momentum with NO corroborating weak signal (volume healthy, price above VWAP, no lower-highs/lower-lows) should NOT downgrade to STALE -- single-signal noise still falls through to WATCH, unchanged from before this feature.)rr r!g\( @r"Q @gGz @gq= ף @)r$RiUiYi]r%eixir(r)r*r6volume_decliningr7r8Nlower_highs_lower_lowsprice_above_vwapzassert %(py1)sSTALEr-)z1%(py2)s {%(py2)s = %(py0)s.raw_state } != %(py5)sr/r0zzmomentum fading alone (no corroboration) should not produce STALE -- got raw_state=STALE with no other weak signal presentr4r5)rr9r:rEr<rArCrD raw_stater=r>r?r@rB)rFrrrr/rLrMrN @py_format2rGrHrIrJrKrrr3test_decelerating_momentum_alone_does_not_downgrade;sP  <<2 zTTestMomentumFadingClassification.test_decelerating_momentum_alone_does_not_downgradec Csgd}gd}t||}tjd|dd}|jdr|jdr|j}d}||k}|sgtd |fd ||fd tvs?t |rDt |nd t |t |d }t d |jdd|i} t t | d}}}tj}|j}||} | } | sddtvst trt tndt |d tvst |rt |nd t |t | d} t t | d}}} } dSdSdS)zFading momentum CORROBORATED by declining volume should reclassify to STALE -- the core new behavior, and the exact QUBT-style pattern (price still nominally positive, but the move is running out of steam) this feature targets.)rr!rPg{G @g(\!@gHz!@g ףp= !@g !@)r'r&rRrQihBi6r#i(#r(r)r*r6rSrV)==)z1%(py2)s {%(py2)s = %(py0)s.raw_state } == %(py5)sr/r0zFexpected STALE with momentum fading + volume declining, got raw_state=r4r5Nzzassert not %(py7)s {%(py7)s = %(py2)s {%(py2)s = %(py0)s.is_eligible_for_entry }(%(py5)s {%(py5)s = %(py3)s.raw_state }) }r9)r1r2py3r3r5)rr9r:rErWr<r=r>r?r@rArBrCrDis_eligible_for_entry) rFrrrr/rGrHrIrJrK @py_assert6 @py_assert8 @py_format9rrr>test_fading_momentum_with_declining_volume_downgrades_to_staleQsP  z_TestMomentumFadingClassification.test_fading_momentum_with_declining_volume_downgrades_to_stalecCstj}d}||}d}||u}|sMtd|fd||fdtvs&ttr+ttndt|t|t|t|d}dd|i}tt |d }}}}}tj}d }||}d}||u}|std|fd||fdtvs}ttrttndt|t|t|t|d}dd|i}tt |d }}}}}tj}d }||}d }||u}|std|fd||fdtvsttrttndt|t|t|t|d}dd|i}tt |d }}}}}d S) aDirectly confirms the protective mechanism: whatever produces a STALE raw_state via the new momentum path is blocked at the same is_eligible_for_entry() gate every other STALE read uses -- this is what makes the feature more than cosmetic.HEALTHYT)is)z[%(py6)s {%(py6)s = %(py2)s {%(py2)s = %(py0)s.is_eligible_for_entry }(%(py4)s) } is %(py9)sr9)r1r2py4py6py9zassert %(py11)spy11NWATCHrVF) r9r\r<r=r>r?r@rArCrD)rFrGrI @py_assert5r^ @py_assert7 @py_format10 @py_format12rrr.test_stale_from_momentum_is_not_entry_eligiblefszOTestMomentumFadingClassification.test_stale_from_momentum_is_not_entry_eligibleN)__name__ __module__ __qualname__rOrYr`rlrrrrr*s  r__main__z-v)N)__doc__builtinsr>_pytest.assertion.rewrite assertionrewriter<sysosrrrpathinsertdirnameabspath__file__r9rrrmpytestmainrrrrs"&  F