Function make_test_context

cake-core/src/models/common/text_model.rs:688–719

Helper to build a minimal Context for testing create_logits_processor.

(temperature: f64, top_k: Option<usize>, top_p: Option<f64>, seed: u64)

    /// Helper to build a minimal Context for testing create_logits_processor.
    fn make_test_context(temperature: f64, top_k: Option<usize>, top_p: Option<f64>, seed: u64) -> crate::cake::Context {
        use crate::cake::{Context, Topology};
        use crate::Args;
        use candle_core::DType;
        use std::path::PathBuf;
        use std::sync::{Arc, Mutex};

        let args = Args {
            temperature,
            top_k,
            top_p,
            seed,
            ..Args::default()
        };

        Context {
            args,
            dtype: DType::F32,
            topology: Topology::new(),
            data_path: PathBuf::from("/tmp"),
            device: Device::Cpu,
            config: None,
            cache: None,
            var_builder: None,
            text_model_arch: crate::TextModelArch::Llama,
            quant: Arc::new(crate::utils::NoQuantization),
            listener_override: Arc::new(Mutex::new(None)),
            tensor_storage: None,
            layer_devices: None,
            backend: Arc::new(crate::backends::CpuBackend::new()),
        }
    }

    #[test]
    fn test_create_logits_processor_argmax_at_zero_temp() {