diff --git a/benches/benches.rs b/benches/benches.rs index 18d4bec..f61af6e 100644 --- a/benches/benches.rs +++ b/benches/benches.rs @@ -13,9 +13,7 @@ pub fn sieve_of_atkin(c: &mut criterion::Criterion) { } pub fn primes(c: &mut criterion::Criterion) { - c.bench_function("primes", |b| { - b.iter(|| utils::Primes::new(10i64.pow(6)).count()) - }); + c.bench_function("primes", |b| b.iter(|| utils::Primes::new(10i64.pow(6)).count())); } criterion::criterion_group!(benches, is_prime, sieve_of_atkin, primes); diff --git a/src/utils/iterators/primes.rs b/src/utils/iterators/primes.rs index dc92847..b14eeff 100644 --- a/src/utils/iterators/primes.rs +++ b/src/utils/iterators/primes.rs @@ -28,9 +28,7 @@ impl Primes { fn next_other_prime(&mut self) -> Option { loop { - let Some(potential_prime) = self.potential_primes.next() else { - return None; - }; + let potential_prime = self.potential_primes.next()?; match self.lookup.remove(&potential_prime) { None => { self.lookup.insert(potential_prime.pow(2), potential_prime);