fix the short_test_cases loop in the C bindings tests

2 files changed, 4 insertions, 2 deletions

diff --git a/c/blake3_c_rust_bindings/benches/bench.rs b/c/blake3_c_rust_bindings/benches/bench.rs
index c6e821e..119bd20 100644
--- a/c/blake3_c_rust_bindings/benches/bench.rs
+++ b/c/blake3_c_rust_bindings/benches/bench.rs
@@ -251,7 +251,6 @@ fn bench_many_parents_sse2(b: &mut Bencher) {
     );
 }
 
-
 #[bench]
 #[cfg(any(target_arch = "x86", target_arch = "x86_64"))]
 fn bench_many_parents_sse41(b: &mut Bencher) {
diff --git a/c/blake3_c_rust_bindings/src/test.rs b/c/blake3_c_rust_bindings/src/test.rs
index ff4b35d..b989ae9 100644
--- a/c/blake3_c_rust_bindings/src/test.rs
+++ b/c/blake3_c_rust_bindings/src/test.rs
@@ -425,7 +425,10 @@ fn reference_hash(input: &[u8]) -> [u8; OUT_LEN] {
 fn test_compare_update_multiple() {
     // Don't use all the long test cases here, since that's unnecessarily slow
     // in debug mode.
-    let short_test_cases = &TEST_CASES[..10];
+    let mut short_test_cases = TEST_CASES;
+    while *short_test_cases.last().unwrap() > 4 * CHUNK_LEN {
+        short_test_cases = &short_test_cases[..short_test_cases.len() - 1];
+    }
     assert_eq!(*short_test_cases.last().unwrap(), 4 * CHUNK_LEN);
 
     let mut input_buf = [0; 2 * TEST_CASES_MAX];