diff --git a/ml/tests/ml_readiness_validation_tests.rs b/ml/tests/ml_readiness_validation_tests.rs index b9cd9dd38..facf298dd 100644 --- a/ml/tests/ml_readiness_validation_tests.rs +++ b/ml/tests/ml_readiness_validation_tests.rs @@ -20,7 +20,7 @@ use ml::real_data_loader::RealDataLoader; /// Test 1: Load real DBN data and validate integrity #[tokio::test] async fn test_load_real_data() -> Result<()> { - let mut loader = RealDataLoader::new("test_data/real/databento"); + let mut loader = RealDataLoader::new_from_workspace()?; // Load ZN.FUT (Treasury futures - best quality data) let bars = loader.load_symbol_data("ZN.FUT").await?; @@ -63,7 +63,7 @@ async fn test_load_real_data() -> Result<()> { /// Test 2: Extract features and technical indicators #[tokio::test] async fn test_feature_extraction() -> Result<()> { - let mut loader = RealDataLoader::new("test_data/real/databento"); + let mut loader = RealDataLoader::new_from_workspace()?; let bars = loader.load_symbol_data("ZN.FUT").await?; // Extract features @@ -162,7 +162,7 @@ async fn test_end_to_end_ml_pipeline() -> Result<()> { println!("═══════════════════════════════════════════════════════════"); // Step 1: Load data - let mut loader = RealDataLoader::new("test_data/real/databento"); + let mut loader = RealDataLoader::new_from_workspace()?; let bars = loader.load_symbol_data("ZN.FUT").await?; println!("✅ Data loading: {} bars", bars.len()); @@ -233,7 +233,7 @@ async fn test_baseline_model_comparison() -> Result<()> { println!("\n📊 Baseline Model Comparison"); println!("═══════════════════════════════════════════════════════════"); - let mut loader = RealDataLoader::new("test_data/real/databento"); + let mut loader = RealDataLoader::new_from_workspace()?; let bars = loader.load_symbol_data("ZN.FUT").await?; let features = loader.extract_features(&bars)?; @@ -285,7 +285,7 @@ async fn test_multi_symbol_validation() -> Result<()> { println!("\n📋 Multi-Symbol Data Quality Validation"); println!("═══════════════════════════════════════════════════════════"); - let mut loader = RealDataLoader::new("test_data/real/databento"); + let mut loader = RealDataLoader::new_from_workspace()?; // Test multiple symbols let symbols = vec!["ZN.FUT", "6E.FUT"];