From fcd6a538e6b0c56b8c7473396853af387eb61edd Mon Sep 17 00:00:00 2001
From: Niko <niko@niko.lgbt>
Date: Mon, 29 Sep 2025 13:07:15 -0700
Subject: [PATCH] Fix up tests

---
 src/standalone.rs     | 11 -----------
 src/whisper_params.rs | 10 +++++-----
 2 files changed, 5 insertions(+), 16 deletions(-)

diff --git a/src/standalone.rs b/src/standalone.rs
index c090e3c..ee61aba 100644
--- a/src/standalone.rs
+++ b/src/standalone.rs
@@ -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"));
-    }
-}
diff --git a/src/whisper_params.rs b/src/whisper_params.rs
index cc7329a..b51b602 100644
--- a/src/whisper_params.rs
+++ b/src/whisper_params.rs
@@ -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);