Skip to content

Commit b30748f

Browse files
committed
Simplify and format
1 parent 16b205f commit b30748f

File tree

3 files changed

+37
-42
lines changed

3 files changed

+37
-42
lines changed

benches/benches.rs

+1-2
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,8 @@
11
#![feature(test)]
22

3-
extern crate primes;
43
extern crate test;
5-
use primes::PrimeSet;
64

5+
use primes::PrimeSet;
76
use test::Bencher;
87

98
#[bench]

src/lib.rs

+1-1
Original file line numberDiff line numberDiff line change
@@ -165,7 +165,7 @@ impl PrimeSet {
165165
///
166166
/// Note that this only requires primes up to `n.sqrt()` to be generated, and will generate
167167
/// them as necessary on its own.
168-
#[cfg_attr(feature = "cargo-clippy", allow(wrong_self_convention))]
168+
#[cfg_attr(feature = "cargo-clippy", allow(clippy::wrong_self_convention))]
169169
pub fn is_prime(&mut self, n: u64) -> bool {
170170
if n <= 1 {
171171
return false;

tests/basictests.rs

+35-39
Original file line numberDiff line numberDiff line change
@@ -1,18 +1,16 @@
1-
extern crate primes;
2-
3-
use primes::{PrimeSet,is_prime,factors,factors_uniq};
1+
use primes::{factors, factors_uniq, is_prime, PrimeSet};
42

53
#[test]
6-
fn test_iter(){
4+
fn test_iter() {
75
let mut pset = PrimeSet::new();
8-
let first_few = [2u64,3,5,7,11,13,17,19,23];
6+
let first_few = [2u64, 3, 5, 7, 11, 13, 17, 19, 23];
97
for (m, &n) in pset.iter().zip(first_few.iter()) {
108
assert_eq!(m, n);
119
}
1210
}
1311

1412
#[test]
15-
fn test_find(){
13+
fn test_find() {
1614
let mut pset = PrimeSet::new();
1715

1816
// pset is empty, so it needs to generate the primes
@@ -27,16 +25,16 @@ fn test_find(){
2725
{
2826
let plst = pset.list();
2927
let plen = plst.len();
30-
assert_eq!(plen, ix_exp+1);
28+
assert_eq!(plen, ix_exp + 1);
3129

32-
assert_eq!(plst[plen-1], n_exp);
30+
assert_eq!(plst[plen - 1], n_exp);
3331
}
3432

3533
assert_eq!(pset.find_vec(n_exp), Some((ix_exp, n_exp)));
3634
}
3735

3836
#[test]
39-
fn test_primes(){
37+
fn test_primes() {
4038
let mut pset = PrimeSet::new();
4139

4240
// note: some are repeated, because the pset list grows as it goes
@@ -49,24 +47,22 @@ fn test_primes(){
4947
assert!(is_prime(13));
5048
assert!(!pset.is_prime(45));
5149
assert!(!is_prime(45));
52-
assert!(!pset.is_prime(13*13));
53-
assert!(!is_prime(13*13));
50+
assert!(!pset.is_prime(13 * 13));
51+
assert!(!is_prime(13 * 13));
5452
assert!(pset.is_prime(13));
5553
assert!(pset.is_prime(7));
5654
assert!(is_prime(7));
5755
assert!(!pset.is_prime(9));
5856
assert!(!is_prime(9));
5957
assert!(pset.is_prime(5));
6058
assert!(is_prime(5));
61-
6259
assert!(pset.is_prime(954377));
6360
assert!(pset.is_prime(954379));
64-
assert!(!pset.is_prime(954377*954379));
65-
66-
assert!(!is_prime(18409199*18409201));
61+
assert!(!pset.is_prime(954377 * 954379));
62+
63+
assert!(!is_prime(18409199 * 18409201));
6764
assert!(pset.is_prime(18409199));
6865
assert!(pset.is_prime(18409201));
69-
7066
assert!(!pset.is_prime(2147483643));
7167
assert!(pset.is_prime(2147483647));
7268
assert!(!pset.is_prime(2147483649));
@@ -83,40 +79,40 @@ fn test_primes(){
8379
}
8480

8581
#[test]
86-
fn test_factors(){
87-
let mut pset = PrimeSet::new();
88-
89-
let ns = [ (1, vec!()),
90-
(2, vec!(2)),
91-
(3, vec!(3)),
92-
(4, vec!(2,2)),
93-
(5, vec!(5)),
94-
(6, vec!(2,3)),
95-
(9, vec!(3,3)),
96-
(12, vec!(2,2,3)),
97-
(121, vec!(11,11)),
98-
(144, vec!(2,2,2,2,3,3)),
99-
(10_000_000, vec!(2,2,2,2,2,2,2,5,5,5,5,5,5,5)),
100-
(100, vec!(2,2,5,5)),
101-
(121, vec!(11, 11)),
102-
];
82+
fn test_factors() {
83+
let mut pset = PrimeSet::new();
84+
85+
let ns = [
86+
(1, vec![]),
87+
(2, vec![2]),
88+
(3, vec![3]),
89+
(4, vec![2, 2]),
90+
(5, vec![5]),
91+
(6, vec![2, 3]),
92+
(9, vec![3, 3]),
93+
(12, vec![2, 2, 3]),
94+
(121, vec![11, 11]),
95+
(144, vec![2, 2, 2, 2, 3, 3]),
96+
(10_000_000, vec![2, 2, 2, 2, 2, 2, 2, 5, 5, 5, 5, 5, 5, 5]),
97+
(100, vec![2, 2, 5, 5]),
98+
(121, vec![11, 11]),
99+
];
103100

104101
// Test unique factors
105-
for &(n, ref v) in ns.iter(){
102+
for &(n, ref v) in ns.iter() {
106103
println!("{}: {:?}", n, v);
107104
assert_eq!(pset.prime_factors(n), *v);
108105
assert_eq!(factors(n), *v);
109106

110107
let unique_factors = factors_uniq(n);
111108

112109
// Get unique factors from the lists we made above
113-
let mut unique_factors_exp : Vec<u64> = v.iter().map(|&x| {x}).collect();
110+
let mut unique_factors_exp: Vec<u64> = v.iter().map(|&x| x).collect();
114111
unique_factors_exp.dedup();
115112

116113
assert_eq!(unique_factors, unique_factors_exp);
117114
}
118115

119-
pset = PrimeSet::new();
120-
assert_eq!(pset.prime_factors(12), vec!(2,2,3));
121-
122-
}
116+
pset = PrimeSet::new();
117+
assert_eq!(pset.prime_factors(12), vec!(2, 2, 3));
118+
}

0 commit comments

Comments
 (0)