foxhunt/crates/ml/tests/test_quantized_exports.rs

//! Test that all INT8 quantized types are properly exported from ml crate root

// Verify all quantized TFT types are accessible from ml:: root
use ml::{
    QuantizedGatedResidualNetwork, QuantizedLSTMEncoder, QuantizedTemporalAttention,
    QuantizedTemporalFusionTransformer, QuantizedVariableSelectionNetwork,
};

#[test]
fn test_quantized_types_exported() {
    // This test verifies that all INT8 quantized types compile and are accessible
    // from the ml crate root namespace (not just ml::tft::)

    // We don't need to instantiate these types, just verify they're in scope
    let _vsn_type: Option<QuantizedVariableSelectionNetwork> = None;
    let _lstm_type: Option<QuantizedLSTMEncoder> = None;
    let _grn_type: Option<QuantizedGatedResidualNetwork> = None;
    let _attention_type: Option<QuantizedTemporalAttention> = None;
    let _tft_type: Option<QuantizedTemporalFusionTransformer> = None;

    // Success! All types are properly exported and accessible
}

#[test]
fn test_quantized_types_from_tft_module() {
    // Also verify types are accessible from ml::tft:: namespace
    use ml::tft::{
        QuantizedGatedResidualNetwork as GrnQuantized, QuantizedLSTMEncoder as LstmQuantized,
        QuantizedTemporalAttention as AttentionQuantized,
        QuantizedTemporalFusionTransformer as TftQuantized,
        QuantizedVariableSelectionNetwork as VsnQuantized,
    };

    let _vsn: Option<VsnQuantized> = None;
    let _lstm: Option<LstmQuantized> = None;
    let _grn: Option<GrnQuantized> = None;
    let _attention: Option<AttentionQuantized> = None;
    let _tft: Option<TftQuantized> = None;
}

#[test]
fn test_memory_optimization_exports() {
    // Verify memory optimization utilities are also exported
    use ml::memory_optimization::{QuantizationConfig, QuantizationType, Quantizer};

    let _quantizer: Option<Quantizer> = None;
    let _config: Option<QuantizationConfig> = None;
    let _type: Option<QuantizationType> = None;
}