Unit Tests

rust-lang · Jan 19, 2018 · d33cc12 · d33cc12
1 parent 5850f0b
commit d33cc12
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/src/libcore/tests/iter.rs b/src/libcore/tests/iter.rs
@@ -161,6 +161,24 @@ fn test_iterator_step_by() {
     assert_eq!(it.next(), None);
 }
 
+#[test]
+fn test_iterator_step_by_nth() {
+    let mut it = (0..16).step_by(5);
+    assert_eq!(it.nth(0), Some(0));
+    assert_eq!(it.nth(0), Some(5));
+    assert_eq!(it.nth(0), Some(10));
+    assert_eq!(it.nth(0), Some(15));
+    assert_eq!(it.nth(0), None);
+
+    let it = (0..18).step_by(5);
+    assert_eq!(it.clone().nth(0), Some(0));
+    assert_eq!(it.clone().nth(1), Some(5));
+    assert_eq!(it.clone().nth(2), Some(10));
+    assert_eq!(it.clone().nth(3), Some(15));
+    assert_eq!(it.clone().nth(4), None);
+    assert_eq!(it.clone().nth(42), None);
+}
+
 #[test]
 #[should_panic]
 fn test_iterator_step_by_zero() {