foxhunt/crates/ml/tests/hyperopt_paths_test.rs

#![allow(
    clippy::assertions_on_constants,
    clippy::assertions_on_result_states,
    clippy::clone_on_copy,
    clippy::decimal_literal_representation,
    clippy::doc_markdown,
    clippy::empty_line_after_doc_comments,
    clippy::field_reassign_with_default,
    clippy::get_unwrap,
    clippy::identity_op,
    clippy::inconsistent_digit_grouping,
    clippy::indexing_slicing,
    clippy::integer_division,
    clippy::len_zero,
    clippy::let_underscore_must_use,
    clippy::manual_div_ceil,
    clippy::manual_let_else,
    clippy::manual_range_contains,
    clippy::modulo_arithmetic,
    clippy::needless_range_loop,
    clippy::non_ascii_literal,
    clippy::redundant_clone,
    clippy::shadow_reuse,
    clippy::shadow_same,
    clippy::shadow_unrelated,
    clippy::single_match_else,
    clippy::str_to_string,
    clippy::string_slice,
    clippy::tests_outside_test_module,
    clippy::too_many_lines,
    clippy::unnecessary_wraps,
    clippy::unseparated_literal_suffix,
    clippy::use_debug,
    clippy::useless_vec,
    clippy::wildcard_enum_match_arm,
    clippy::else_if_without_else,
    clippy::expect_used,
    clippy::missing_const_for_fn,
    clippy::similar_names,
    clippy::type_complexity,
    clippy::collapsible_else_if,
    clippy::doc_lazy_continuation,
    clippy::items_after_test_module,
    clippy::map_clone,
    clippy::multiple_unsafe_ops_per_block,
    clippy::unwrap_or_default,
    clippy::assign_op_pattern,
    clippy::needless_borrow,
    clippy::println_empty_string,
    clippy::unnecessary_cast,
    clippy::used_underscore_binding,
    clippy::create_dir,
    clippy::implicit_saturating_sub,
    clippy::exit,
    clippy::expect_fun_call,
    clippy::too_many_arguments,
    clippy::unnecessary_map_or,
    clippy::unwrap_used,
    dead_code,
    unused_imports,
    unused_variables,
    clippy::cloned_ref_to_slice_refs,
    clippy::neg_multiply,
    clippy::while_let_loop,
    clippy::bool_assert_comparison,
    clippy::excessive_precision,
    clippy::trivially_copy_pass_by_ref,
    clippy::op_ref,
    clippy::redundant_closure,
    clippy::unnecessary_lazy_evaluations,
    clippy::if_then_some_else_none,
    clippy::unnecessary_to_owned,
    clippy::single_component_path_imports,
)]
use ml::hyperopt::paths::{generate_run_id, TrainingPaths};
use tempfile::TempDir;

#[test]
fn test_paths_creation_and_cleanup() {
    let temp_dir = TempDir::new().unwrap();
    let paths = TrainingPaths::new(temp_dir.path(), "test_model", "test_run");

    // Create directories
    paths.create_all().unwrap();

    // Verify all directories exist
    assert!(paths.checkpoints_dir().exists());
    assert!(paths.logs_dir().exists());
    assert!(paths.hyperopt_dir().exists());
    assert!(paths.metrics_dir().exists());
}

#[test]
fn test_run_id_uniqueness() {
    let id1 = generate_run_id("test");
    std::thread::sleep(std::time::Duration::from_secs(1));
    let id2 = generate_run_id("test");
    assert_ne!(id1, id2);
}

#[test]
fn test_path_structure() {
    let temp_dir = TempDir::new().unwrap();
    let paths = TrainingPaths::new(temp_dir.path(), "mamba2", "20251028_223000_hyperopt");

    // Verify path structure
    let expected_run_dir = temp_dir
        .path()
        .join("training_runs/mamba2/run_20251028_223000_hyperopt");
    assert_eq!(paths.run_dir(), expected_run_dir);

    let expected_checkpoints = expected_run_dir.join("checkpoints");
    assert_eq!(paths.checkpoints_dir(), expected_checkpoints);

    let expected_logs = expected_run_dir.join("logs");
    assert_eq!(paths.logs_dir(), expected_logs);

    let expected_hyperopt = expected_run_dir.join("hyperopt");
    assert_eq!(paths.hyperopt_dir(), expected_hyperopt);

    let expected_metrics = expected_run_dir.join("metrics");
    assert_eq!(paths.metrics_dir(), expected_metrics);
}

#[test]
fn test_multiple_models_isolation() {
    let temp_dir = TempDir::new().unwrap();

    let paths_mamba2 = TrainingPaths::new(temp_dir.path(), "mamba2", "run1");
    let paths_tft = TrainingPaths::new(temp_dir.path(), "tft", "run1");

    paths_mamba2.create_all().unwrap();
    paths_tft.create_all().unwrap();

    // Verify models are isolated
    assert_ne!(paths_mamba2.run_dir(), paths_tft.run_dir());
    assert!(paths_mamba2.run_dir().to_str().unwrap().contains("mamba2"));
    assert!(paths_tft.run_dir().to_str().unwrap().contains("tft"));
}

#[test]
fn test_run_id_format() {
    let run_id = generate_run_id("hyperopt");

    // Should have format: YYYYMMDD_HHMMSS_hyperopt
    let parts: Vec<&str> = run_id.split('_').collect();
    assert_eq!(parts.len(), 3);

    // Date part should be 8 digits
    assert_eq!(parts[0].len(), 8);
    assert!(parts[0].chars().all(|c| c.is_ascii_digit()));

    // Time part should be 6 digits
    assert_eq!(parts[1].len(), 6);
    assert!(parts[1].chars().all(|c| c.is_ascii_digit()));

    // Type part should match input
    assert_eq!(parts[2], "hyperopt");
}

#[test]
fn test_no_hardcoded_paths() {
    // Test that paths are fully configurable
    let custom_base = "/custom/path";
    let custom_model = "custom_model";
    let custom_run = "custom_run_id";

    let paths = TrainingPaths::new(custom_base, custom_model, custom_run);

    // All paths should start with custom base
    assert!(paths.run_dir().starts_with(custom_base));
    assert!(paths.checkpoints_dir().starts_with(custom_base));
    assert!(paths.logs_dir().starts_with(custom_base));
    assert!(paths.hyperopt_dir().starts_with(custom_base));
    assert!(paths.metrics_dir().starts_with(custom_base));

    // All paths should contain custom model name
    assert!(paths.run_dir().to_str().unwrap().contains(custom_model));

    // All paths should contain custom run id
    assert!(paths.run_dir().to_str().unwrap().contains(custom_run));
}