o سj @sdZddlZddlmmZddlZddlZddl Z ddl m Z m Z m Z ej dej ej ej eddlmZmZmZddlmZmZddlmZddZd d ZGd d d e jZed krke dSdS)aJ test_persistence_gap_bugfix.py Regression test for the "stale persistence survives an evaluation gap" bug found in production logs on 2026-09-14. fast_entry_gate.py's persistence_state only ever stored {"since": timestamp}, and every call just computed `now - since` with no check that the symbol had actually been re-evaluated on every poll in between. Three real, distinct causes of a gap all produced the same result -- a BUY firing off a stale "since" with zero continuous re-confirmation across the gap: - ALLT: direction flipped UP at 09:49:25, got benched (5 consecutive non-confirms -> 180s off the shortlist) at 09:49:35, then bought instantly at 09:53:21 on the very first re-check (held=236.2s). - JBS: direction flipped UP at 09:50:08, health dropped to STALE at 09:50:16 (dropping it out of the health-eligible pool entirely), then bought instantly at 09:53:28 once health recovered (held=199.5s). - FUBO: carried a since-timestamp across a 2h26m (8804.6s) health-STALE gap and bought the instant health recovered, then repeated a rapid-refire pattern (bought/exited/re-bought within seconds, since being open/closing also excludes a symbol from re-evaluation without clearing its persistence timer) -- same root cause as BTE's 5 back-to-back buy/momentum_fade_exit cycles in under a minute. Fix: persistence_state also tracks "last_seen" (the as_of of the last call that reached the UP-tracking branch). If the gap between "now" and "last_seen" exceeds max_evaluation_gap_seconds (config default 15s -- enough to tolerate one or two missed ~5s poll ticks, not a 180s bench or a multi-minute/hour health gap), the old "since" is discarded as stale and persistence restarts from zero. Run: python -m pytest tests/test_persistence_gap_bugfix.py -v or: python tests/test_persistence_gap_bugfix.py N)datetime timedeltatimezone)evaluate_fast_entry STATE_BUY STATE_WAITFastPredictionReading DIRECTION_UPStreamFeaturesc Cstdtddddddddgd S) NTESTgT@StrongSteadyBullishSAFENORMALzmomentum=Strong) symbol direction confidencemomentum volume_state vwap_state ema9_state resistance extensionreasonsrrr@/var/www/screener/premarket/tests/test_persistence_gap_bugfix.py_up_prediction3s rc Cs,td|ddiddiddidd d d d id S) Nr last$@momentum_accelerationg?relative_volumeg@g#@g?)valueslope spread_pct)r computed_atprice accelerationvolumevwapquoter as_ofrrr_clean_features<sr/c@s$eZdZddZddZddZdS)TestPersistenceGapBugfixc Cst}tddddddtjd}i}d}tdD]}|td|d }td |t|||d }|j}q| |j t | |j dS) zSanity check: normal back-to-back ~5s polls with no gap should still accumulate persistence and fire BUY once >= 18s, unaffected by the gap-detection fix. 1tzinfoNsecondsr r-)rrrutcrangerrr/ persistence assertEqualstater assertTrue should_enter)self predictiont0r?decisioninowrrr>test_continuous_evaluation_still_buys_after_persistence_windowHs zWTestPersistenceGapBugfix.test_continuous_evaluation_still_buys_after_persistence_windowc Cst}tddddddtjd}i}td|t|||d}|j}||jt |t d d }td|t|||d}||jt d | |j d d dS)aReproduces the ALLT case: UP flips, then a long gap (simulating a 180s bench) where the symbol isn't evaluated at all, then it comes back still UP. Before the fix this fired BUY instantly off the stale "since". After the fix it must restart from zero.r1r2r3r4r5r6r r-gfffffm@r9zha since-timestamp that survived a long unevaluated gap must not instantly satisfy the persistence windowg?z>persistence must restart from ~0s, not carry over the 236s gapN) rrrr;rr/r=r>r?rr assertLesspersistence_seconds_elapsed)rBrC t_up_startr?rE t_after_benchrrr)test_gap_from_benching_resets_persistenceWs  zBTestPersistenceGapBugfix.test_gap_from_benching_resets_persistencec Csxt}tddddddtjd}i}td|t|||d}|j}|td d }td|t|||d}|j|j d d d dS)zA single missed poll tick (e.g. a slow cycle) shouldn't punish a genuinely continuous setup -- only gaps beyond max_evaluation_gap_seconds (15s default) should reset.r1r2r3r4r5r6r r- r9r!g{Gz?)deltaN) rrrr;rr/r=rassertAlmostEqualrJ)rBrCrDr?rEt1rrr.test_short_gap_within_tolerance_does_not_resetpszGTestPersistenceGapBugfix.test_short_gap_within_tolerance_does_not_resetN)__name__ __module__ __qualname__rHrMrRrrrrr0Gs r0__main__)!__doc__builtins @py_builtins_pytest.assertion.rewrite assertionrewrite @pytest_arsysosunittestrrrpathinsertdirnameabspath__file__fast_entry_gaterrrfast_prediction_enginer r stream_featuresr rr/TestCaser0rSmainrrrrs"&&   9