Add unit test for prime_counts

JSorngard · Oct 22, 2023 · 4939b52 · 4939b52
1 parent f0d010e
commit 4939b52
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -532,6 +532,22 @@ mod test {
         );
     }
 
+    #[test]
+    fn check_prime_count() {
+        const N: usize = 100;
+        const PRIME_COUNTS: [usize; N] = prime_counts();
+        let mut counts = [0; N];
+        for n in 0..N {
+            let mut i = 0;
+            while PRECOMPUTED_PRIMES[i] <= n as u32 {
+                counts[n] += 1;
+                i += 1;
+            }
+        }
+
+        assert_eq!(PRIME_COUNTS, counts);
+    }
+
     #[test]
     fn check_möbius() {
         #[rustfmt::skip]