repeat iter: tests + make map last work

Author: hkBst

library/core/src/iter/adapters/map.rs

@@ -112,6 +112,13 @@ where
  self.iter.size_hint()
  }
 
+ fn last(mut self) -> Option<Self::Item>
+ where
+ Self: Sized,
+ {
+ self.iter.last().map(&mut self.f)
+ }
+
  fn try_fold<Acc, G, R>(&mut self, init: Acc, g: G) -> R
  where
  Self: Sized,

library/coretests/tests/iter/sources.rs

@@ -30,6 +30,22 @@ fn test_repeat_take_collect() {
  assert_eq!(v, vec![42, 42, 42]);
 }
 
+#[test]
+#[should_panic = "iterator is infinite"]
+fn test_repeat_count() {
+ repeat(42).count();
+}
+
+#[test]
+fn test_repeat_last() {
+ assert_eq!(repeat(42).last(), Some(42));
+}
+
+#[test]
+fn test_repeat_map_double_last() {
+ assert_eq!(repeat(42).map(|e| 2 * e).last(), Some(2 * 42));
+}
+
 #[test]
 fn test_repeat_with() {
  #[derive(PartialEq, Debug)]