Skip to content

Commit ffb1ef5

Browse files
xuganyu96fjarri
authored andcommitted
Add tests for BoxedUint
1 parent da7a174 commit ffb1ef5

File tree

1 file changed

+19
-1
lines changed

1 file changed

+19
-1
lines changed

src/presets.rs

Lines changed: 19 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -192,7 +192,7 @@ fn _is_prime_with_rng<T: Integer + RandomMod>(rng: &mut impl CryptoRngCore, num:
192192

193193
#[cfg(test)]
194194
mod tests {
195-
use crypto_bigint::{CheckedAdd, Uint, Word, U128, U64};
195+
use crypto_bigint::{BoxedUint, CheckedAdd, Uint, Word, U128, U64};
196196
use num_prime::nt_funcs::is_prime64;
197197
use rand_core::OsRng;
198198

@@ -279,6 +279,15 @@ mod tests {
279279
}
280280
}
281281

282+
#[test]
283+
fn prime_generation_boxed() {
284+
for bit_length in (28..=128).step_by(10) {
285+
let p: BoxedUint = generate_prime(bit_length, 128);
286+
assert!(p.bits_vartime() == bit_length);
287+
assert!(is_prime(&p));
288+
}
289+
}
290+
282291
#[test]
283292
fn safe_prime_generation() {
284293
for bit_length in (28..=128).step_by(10) {
@@ -288,6 +297,15 @@ mod tests {
288297
}
289298
}
290299

300+
#[test]
301+
fn safe_prime_generation_boxed() {
302+
for bit_length in (28..=128).step_by(10) {
303+
let p: BoxedUint = generate_safe_prime(bit_length, 128);
304+
assert!(p.bits_vartime() == bit_length);
305+
assert!(is_safe_prime(&p));
306+
}
307+
}
308+
291309
#[test]
292310
fn corner_cases_is_prime() {
293311
for num in 0u64..30 {

0 commit comments

Comments
 (0)