#![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)); }