better pivots

test_pivot_levels.py

@@ -0,0 +1,84 @@
+#!/usr/bin/env python3
+"""
+Test script to verify all 5 pivot levels are being calculated
+"""
+
+import sys
+import os
+import logging
+
+# Add the project root to the Python path
+sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
+from core.data_provider import DataProvider
+
+# Configure logging
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
+)
+logger = logging.getLogger(__name__)
+
+def test_pivot_levels():
+    """Test all 5 pivot levels calculation"""
+    try:
+        logger.info("Initializing DataProvider...")
+        data_provider = DataProvider()
+
+        # Wait for initial data to load
+        import time
+        time.sleep(3)
+
+        # Test pivot levels for ETH/USDT
+        symbol = 'ETH/USDT'
+        logger.info(f"\nTesting Williams pivot levels for {symbol}:")
+
+        # Get pivot levels
+        pivot_levels = data_provider.get_williams_pivot_levels(symbol, base_timeframe='1m', limit=5000)
+
+        if not pivot_levels:
+            logger.error(f"❌ NO PIVOT LEVELS for {symbol}")
+            return False
+
+        logger.info(f"✅ Found {len(pivot_levels)} pivot levels")
+
+        for level_num in sorted(pivot_levels.keys()):
+            trend_level = pivot_levels[level_num]
+            pivot_count = len(getattr(trend_level, 'pivot_points', []))
+            direction = getattr(trend_level, 'trend_direction', 'unknown')
+            strength = getattr(trend_level, 'trend_strength', 0.0)
+
+            logger.info(f"  Level {level_num}: {pivot_count} pivots, {direction} ({strength:.1%})")
+
+            if pivot_count > 0:
+                # Show sample of pivot types
+                high_count = sum(1 for p in trend_level.pivot_points if getattr(p, 'pivot_type', '') == 'high')
+                low_count = sum(1 for p in trend_level.pivot_points if getattr(p, 'pivot_type', '') == 'low')
+                logger.info(f"    High pivots: {high_count}, Low pivots: {low_count}")
+
+        # Check if we have all levels
+        expected_levels = {1, 2, 3, 4, 5}
+        actual_levels = set(pivot_levels.keys())
+
+        if expected_levels.issubset(actual_levels):
+            logger.info("✅ ALL 5 PIVOT LEVELS PRESENT!")
+        else:
+            missing = expected_levels - actual_levels
+            logger.warning(f"❌ MISSING LEVELS: {missing}")
+
+        return True
+
+    except Exception as e:
+        logger.error(f"Test failed with error: {e}")
+        import traceback
+        traceback.print_exc()
+        return False
+
+if __name__ == "__main__":
+    success = test_pivot_levels()
+    if success:
+        print("\n🎉 Pivot levels test completed!")
+    else:
+        print("\n❌ Pivot levels test failed!")
+        sys.exit(1)
+