Files
XauBot/tests/test_profit_momentum.py
T
GifariKemalandClaude Sonnet 4.5 ecfe3615ac docs: add profit momentum research artifacts
Added profit momentum feature research files from previous analysis:
- docs/research/PROFIT_MOMENTUM_CODE_SNIPPET.py — Implementation code
- docs/research/PROFIT_MOMENTUM_INTEGRATION.md — Integration guide
- tests/test_profit_momentum.py — Test script

These files document profit momentum feature exploration (unrelated to
current HMM fix but kept for reference).

Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
2026-02-09 10:52:05 +07:00

264 lines
7.8 KiB
Python

"""
Test Profit Momentum Tracker
=============================
Demo dan test untuk profit momentum tracking system.
"""
import sys
import os
import time
import random
from pathlib import Path
# Add project root to path
project_root = Path(__file__).parent.parent
sys.path.insert(0, str(project_root))
from src.profit_momentum_tracker import ProfitMomentumTracker
from loguru import logger
def simulate_profit_pattern_1():
"""
Simulate Pattern 1: Steady Growth then Reversal
- Profit grows steadily
- Peaks at $50
- Then reverses slowly
Expected: Should exit around $45-$47 (90-94% of peak)
"""
logger.info("=" * 60)
logger.info("PATTERN 1: Steady Growth → Reversal")
logger.info("=" * 60)
tracker = ProfitMomentumTracker(
enable_logging=True,
min_profit_for_momentum_exit=5.0,
grace_period_seconds=3.0,
)
ticket = 123456
price = 2650.0
# Phase 1: Steady growth (0-10s)
logger.info("\n📈 Phase 1: Steady Growth (0-10s)")
for i in range(20): # 10 seconds at 500ms interval
profit = i * 2.5 # Linear growth to $50
price += 0.5
tracker.update(ticket, profit, price)
time.sleep(0.5)
if i % 4 == 0: # Log every 2 seconds
metrics = tracker.calculate_metrics(ticket)
if metrics:
logger.info(
f" t={i*0.5:.1f}s | Profit: ${profit:.2f} | "
f"Velocity: {metrics.velocity:.2f} $/s | "
f"Momentum: {metrics.momentum_direction}"
)
# Phase 2: Peak stagnation (10-13s)
logger.info("\n⏸️ Phase 2: Peak Stagnation (10-13s)")
for i in range(6): # 3 seconds
profit = 50.0 + random.uniform(-0.5, 0.5) # Stagnant around $50
price += random.uniform(-0.1, 0.1)
tracker.update(ticket, profit, price)
should_exit, reason = tracker.should_exit(ticket, profit)
if should_exit:
logger.warning(f"🚨 EXIT SIGNAL: {reason}")
break
time.sleep(0.5)
# Phase 3: Slow reversal (13-20s)
logger.info("\n📉 Phase 3: Slow Reversal (13-20s)")
for i in range(14): # 7 seconds
profit = 50.0 - (i * 1.5) # Decline from $50
price -= 0.3
tracker.update(ticket, profit, price)
should_exit, reason = tracker.should_exit(ticket, profit)
metrics = tracker.calculate_metrics(ticket)
if metrics and i % 2 == 0:
logger.info(
f" t={13+i*0.5:.1f}s | Profit: ${profit:.2f} | "
f"Velocity: {metrics.velocity:.2f} $/s | "
f"Peak Drawdown: {metrics.drawdown_from_peak:.1f}%"
)
if should_exit:
logger.warning(f"🚨 EXIT SIGNAL at ${profit:.2f}: {reason}")
summary = tracker.get_position_summary(ticket)
logger.success(
f"✅ Exit Summary: Peak ${summary['peak_profit']:.2f} → "
f"Exit ${profit:.2f} ({summary['drawdown_pct']:.1f}% from peak)"
)
break
time.sleep(0.5)
def simulate_profit_pattern_2():
"""
Simulate Pattern 2: Quick Spike then Sharp Reversal
- Profit spikes quickly to $40
- Reverses sharply
Expected: Should exit quickly on velocity reversal
"""
logger.info("\n" + "=" * 60)
logger.info("PATTERN 2: Quick Spike → Sharp Reversal")
logger.info("=" * 60)
tracker = ProfitMomentumTracker(
enable_logging=True,
velocity_reversal_threshold=-1.0, # More sensitive
min_profit_for_momentum_exit=5.0,
)
ticket = 234567
price = 2650.0
# Phase 1: Quick spike (0-4s)
logger.info("\n🚀 Phase 1: Quick Spike (0-4s)")
for i in range(8): # 4 seconds
profit = i * 5.0 # Fast growth to $40
price += 1.0
tracker.update(ticket, profit, price)
time.sleep(0.5)
metrics = tracker.calculate_metrics(ticket)
if metrics and i % 2 == 0:
logger.info(
f" t={i*0.5:.1f}s | Profit: ${profit:.2f} | "
f"Velocity: {metrics.velocity:.2f} $/s"
)
# Phase 2: Sharp reversal (4-8s)
logger.info("\n💥 Phase 2: Sharp Reversal (4-8s)")
for i in range(8): # 4 seconds
profit = 40.0 - (i * 4.0) # Fast decline
price -= 0.8
tracker.update(ticket, profit, price)
should_exit, reason = tracker.should_exit(ticket, profit)
metrics = tracker.calculate_metrics(ticket)
if metrics:
logger.info(
f" t={4+i*0.5:.1f}s | Profit: ${profit:.2f} | "
f"Velocity: {metrics.velocity:.2f} $/s | "
f"Accel: {metrics.acceleration:.2f} $/s²"
)
if should_exit:
logger.warning(f"🚨 EXIT SIGNAL at ${profit:.2f}: {reason}")
summary = tracker.get_position_summary(ticket)
logger.success(
f"✅ Exit Summary: Peak ${summary['peak_profit']:.2f} → "
f"Exit ${profit:.2f}"
)
break
time.sleep(0.5)
def simulate_profit_pattern_3():
"""
Simulate Pattern 3: Healthy Trend (No Exit)
- Profit grows steadily
- Small pullbacks but momentum stays positive
Expected: Should NOT exit (healthy momentum)
"""
logger.info("\n" + "=" * 60)
logger.info("PATTERN 3: Healthy Trend (No Exit Expected)")
logger.info("=" * 60)
tracker = ProfitMomentumTracker(
enable_logging=True,
peak_drawdown_threshold=50.0, # Allow larger drawdown
)
ticket = 345678
price = 2650.0
# Simulate 15 seconds of healthy growth with small pullbacks
logger.info("\n📊 Simulating healthy trend with pullbacks...")
for i in range(30): # 15 seconds
# Add some volatility but overall uptrend
base_profit = i * 1.5
noise = random.uniform(-2.0, 3.0) # Slight upward bias
profit = base_profit + noise
price += random.uniform(-0.2, 0.5)
tracker.update(ticket, profit, price)
should_exit, reason = tracker.should_exit(ticket, profit)
if i % 4 == 0: # Log every 2 seconds
metrics = tracker.calculate_metrics(ticket)
if metrics:
logger.info(
f" t={i*0.5:.1f}s | Profit: ${profit:.2f} | "
f"Peak: ${metrics.peak_profit:.2f} | "
f"Velocity: {metrics.velocity:.2f} $/s | "
f"Status: {metrics.momentum_direction}"
)
if should_exit:
logger.warning(f"⚠️ Unexpected exit: {reason}")
break
time.sleep(0.5)
if not should_exit:
logger.success("✅ No exit triggered - Healthy trend maintained!")
summary = tracker.get_position_summary(ticket)
if summary:
logger.info(
f"Final Stats: Peak ${summary['peak_profit']:.2f}, "
f"Current ${summary['current_profit']:.2f}, "
f"Velocity {summary['velocity']:.2f} $/s"
)
def main():
"""Run all simulation patterns."""
logger.remove()
logger.add(
sys.stdout,
format="<green>{time:HH:mm:ss.SSS}</green> | <level>{message}</level>",
level="INFO",
)
logger.info("🧪 Profit Momentum Tracker - Simulation Tests")
logger.info("=" * 60)
try:
# Run pattern simulations
simulate_profit_pattern_1()
time.sleep(2)
simulate_profit_pattern_2()
time.sleep(2)
simulate_profit_pattern_3()
except KeyboardInterrupt:
logger.warning("\n⚠️ Simulation interrupted by user")
except Exception as e:
logger.error(f"❌ Error: {e}", exc_info=True)
logger.info("\n" + "=" * 60)
logger.success("🎉 All simulations completed!")
if __name__ == "__main__":
main()