harald/wyoming-whisper-rs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix up tests

Browse source
This commit is contained in:
Niko 2025-09-29 13:07:15 -07:00
parent 2fa5df779c
commit fcd6a538e6
No known key found for this signature in database
2 changed files with 5 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

11
src/standalone.rs
View file

@ -128,14 +128,3 @@ impl Default for SystemInfo {
}
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_openblas() {
let info = SystemInfo::default();
assert_eq!(info.blas, cfg!(feature = "openblas"));
}
}

10
src/whisper_params.rs
View file

@ -868,7 +868,7 @@ mod test_whisper_params_initial_prompt {
#[test]
fn test_initial_prompt_normal_usage() {
let mut params = FullParams::new(SamplingStrategy::default());
let mut params = FullParams::new(SamplingStrategy::Greedy { best_of: 5 });
let prompt = "Hello, world!";
params.set_initial_prompt(prompt);
assert_eq!(params.get_initial_prompt(), prompt);
@ -877,7 +877,7 @@ mod test_whisper_params_initial_prompt {
#[test]
#[should_panic(expected = "Initial prompt contains null byte")]
fn test_initial_prompt_null_byte() {
let mut params = FullParams::new(SamplingStrategy::default());
let mut params = FullParams::new(SamplingStrategy::Greedy { best_of: 5 });
let prompt = "Hello\0, world!";
params.set_initial_prompt(prompt);
// Should panic
@ -885,7 +885,7 @@ mod test_whisper_params_initial_prompt {
#[test]
fn test_initial_prompt_empty_string() {
let mut params = FullParams::new(SamplingStrategy::default());
let mut params = FullParams::new(SamplingStrategy::Greedy { best_of: 5 });
let prompt = "";
params.set_initial_prompt(prompt);
@ -898,7 +898,7 @@ mod test_whisper_params_initial_prompt {
#[test]
fn test_initial_prompt_repeated_calls() {
let mut params = FullParams::new(SamplingStrategy::default());
let mut params = FullParams::new(SamplingStrategy::Greedy { best_of: 5 });
params.set_initial_prompt("First prompt");
assert_eq!(
params.get_initial_prompt(),
@ -916,7 +916,7 @@ mod test_whisper_params_initial_prompt {
#[test]
fn test_initial_prompt_long_string() {
let mut params = FullParams::new(SamplingStrategy::default());
let mut params = FullParams::new(SamplingStrategy::Greedy { best_of: 5 });
let long_prompt = "a".repeat(10000); // a long string of 10,000 'a' characters
params.set_initial_prompt(&long_prompt);