Add tests of sieve_lt

Author: JSorngard

src/sieve.rs

@@ -423,9 +423,9 @@ macro_rules! sieve_segment {
 
 #[cfg(test)]
 mod test {
-    use crate::sieve::SegmentedSieveError;
+    use crate::SieveError;
 
-    use super::{sieve, sieve_segment};
+    use super::{sieve, sieve_lt, sieve_segment, SegmentedSieveError};
 
     #[test]
     fn test_consistency_of_sieve_segment() {
@@ -445,4 +445,12 @@ mod test {
         );
         assert_eq!(sieve_segment(&sieve::<5>(), 5), Ok(sieve()));
     }
+
+    #[test]
+    fn test_sieve_lt() {
+        assert_eq!(sieve_lt::<5, 5>(30), Err(SieveError::TooSmallSieveSize));
+        assert_eq!(sieve_lt::<5, 5>(4), Err(SieveError::TooSmallLimit));
+        assert_eq!(sieve_lt::<5, 5>(5), Ok(sieve()));
+        assert_eq!(sieve_lt::<2, 5>(20), Ok([false, true]));
+    }
 }