Skip to content

Commit 4019e41

Browse files
committed
fn ZDICT_fillNoise: test
1 parent 4e408c1 commit 4019e41

File tree

1 file changed

+14
-4
lines changed

1 file changed

+14
-4
lines changed

lib/dictBuilder/zdict.rs

Lines changed: 14 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -716,15 +716,15 @@ unsafe fn ZDICT_trainBuffer_legacy(
716716
result
717717
}
718718

719-
fn ZDICT_fillNoise(buffer: &mut [u8]) {
719+
fn fill_noise(buffer: &mut [u8]) {
720720
const prime1: u32 = 2654435761;
721721
const prime2: u32 = 2246822519;
722722

723723
let mut acc = prime1;
724724

725-
for e in buffer.iter_mut() {
725+
for v in buffer.iter_mut() {
726726
acc = acc.wrapping_mul(prime2);
727-
*e = (acc >> 21) as u8;
727+
*v = (acc >> 21) as u8;
728728
}
729729
}
730730

@@ -1576,7 +1576,7 @@ pub unsafe extern "C" fn ZDICT_trainFromBuffer_legacy(
15761576
}
15771577
let mut new_buf = vec![0u8; sBuffSize.wrapping_add(NOISELENGTH as size_t)];
15781578
core::ptr::copy_nonoverlapping(samplesBuffer.cast::<u8>(), new_buf.as_mut_ptr(), sBuffSize);
1579-
ZDICT_fillNoise(&mut new_buf[sBuffSize..]);
1579+
fill_noise(&mut new_buf[sBuffSize..]);
15801580
ZDICT_trainFromBuffer_unsafe_legacy(
15811581
dictBuffer,
15821582
dictBufferCapacity,
@@ -1675,4 +1675,14 @@ mod test {
16751675
let code = unsafe { ZDICT_getDictID(DICT.as_ptr().cast(), DICT.len()) };
16761676
assert_eq!(code, 1877512422);
16771677
}
1678+
1679+
#[test]
1680+
fn test_fill_noise() {
1681+
let mut buf = vec![0u8; 16];
1682+
fill_noise(&mut buf);
1683+
assert_eq!(
1684+
buf,
1685+
[226, 51, 247, 105, 221, 225, 137, 112, 5, 188, 15, 79, 183, 243, 110, 209]
1686+
);
1687+
}
16781688
}

0 commit comments

Comments
 (0)