RGBCube/uutils-coreutils
1
Fork 0
mirror of https://github.com/RGBCube/uutils-coreutils synced 2025-08-02 05:57:46 +00:00
Code Issues Activity

factor::miller_rabin::tests: Refactor

Browse source
This commit is contained in:
nicoo 2020-07-04 20:35:16 +02:00
parent 0a1200bdb8
commit 600268c6e4
1 changed files with 12 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

21
src/uu/factor/src/miller_rabin.rs
View file

@ -116,6 +116,14 @@ mod tests {
use super::is_prime; use super::is_prime;
const LARGEST_U64_PRIME: u64 = 0xFFFFFFFFFFFFFFC5; const LARGEST_U64_PRIME: u64 = 0xFFFFFFFFFFFFFFC5;
fn primes() -> impl Iterator<Item = u64> {
use crate::table::{NEXT_PRIME, P_INVS_U64};
use std::iter::once;
once(2)
.chain(P_INVS_U64.iter().map(|(p, _, _)| *p))
.chain(once(NEXT_PRIME))
}
#[test] #[test]
fn largest_prime() { fn largest_prime() {
assert!(is_prime(LARGEST_U64_PRIME)); assert!(is_prime(LARGEST_U64_PRIME));
@ -130,11 +138,9 @@ mod tests {
#[test] #[test]
fn first_primes() { fn first_primes() {
use crate::table::{NEXT_PRIME, P_INVS_U64}; for p in primes() {
for (p, _, _) in P_INVS_U64.iter() { assert!(is_prime(p), "{} reported composite", p);
assert!(is_prime(*p), "{} reported composite", p);
} }
assert!(is_prime(NEXT_PRIME));
} }
#[test] #[test]
@ -145,11 +151,8 @@ mod tests {
#[test] #[test]
fn small_composites() { fn small_composites() {
use crate::table::P_INVS_U64; for p in primes() {
for q in primes().take_while(|q| *q <= p) {
for i in 0..P_INVS_U64.len() {
let (p, _, _) = P_INVS_U64[i];
for (q, _, _) in &P_INVS_U64[0..i] {
let n = p * q; let n = p * q;
assert!(!is_prime(n), "{} = {} × {} reported prime", n, p, q); assert!(!is_prime(n), "{} = {} × {} reported prime", n, p, q);
} }