src/libcore/tests/iter.rs

   1 // Copyright 2014 The Rust Project Developers. See the COPYRIGHT
   2 // file at the top-level directory of this distribution and at
   3 // http://rust-lang.org/COPYRIGHT.
   4 //
   5 // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
   6 // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
   7 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
   8 // option. This file may not be copied, modified, or distributed
   9 // except according to those terms.
  10
  11 use core::iter::*;
  12 use core::{i8, i16, isize};
  13 use core::usize;
  14
  15 #[test]
  16 fn test_lt() {
  17     let empty: [isize; 0] = [];
  18     let xs = [1,2,3];
  19     let ys = [1,2,0];
  20
  21     assert!(!xs.iter().lt(ys.iter()));
  22     assert!(!xs.iter().le(ys.iter()));
  23     assert!( xs.iter().gt(ys.iter()));
  24     assert!( xs.iter().ge(ys.iter()));
  25
  26     assert!( ys.iter().lt(xs.iter()));
  27     assert!( ys.iter().le(xs.iter()));
  28     assert!(!ys.iter().gt(xs.iter()));
  29     assert!(!ys.iter().ge(xs.iter()));
  30
  31     assert!( empty.iter().lt(xs.iter()));
  32     assert!( empty.iter().le(xs.iter()));
  33     assert!(!empty.iter().gt(xs.iter()));
  34     assert!(!empty.iter().ge(xs.iter()));
  35
  36     // Sequence with NaN
  37     let u = [1.0f64, 2.0];
  38     let v = [0.0f64/0.0, 3.0];
  39
  40     assert!(!u.iter().lt(v.iter()));
  41     assert!(!u.iter().le(v.iter()));
  42     assert!(!u.iter().gt(v.iter()));
  43     assert!(!u.iter().ge(v.iter()));
  44
  45     let a = [0.0f64/0.0];
  46     let b = [1.0f64];
  47     let c = [2.0f64];
  48
  49     assert!(a.iter().lt(b.iter()) == (a[0] <  b[0]));
  50     assert!(a.iter().le(b.iter()) == (a[0] <= b[0]));
  51     assert!(a.iter().gt(b.iter()) == (a[0] >  b[0]));
  52     assert!(a.iter().ge(b.iter()) == (a[0] >= b[0]));
  53
  54     assert!(c.iter().lt(b.iter()) == (c[0] <  b[0]));
  55     assert!(c.iter().le(b.iter()) == (c[0] <= b[0]));
  56     assert!(c.iter().gt(b.iter()) == (c[0] >  b[0]));
  57     assert!(c.iter().ge(b.iter()) == (c[0] >= b[0]));
  58 }
  59
  60 #[test]
  61 fn test_multi_iter() {
  62     let xs = [1,2,3,4];
  63     let ys = [4,3,2,1];
  64     assert!(xs.iter().eq(ys.iter().rev()));
  65     assert!(xs.iter().lt(xs.iter().skip(2)));
  66 }
  67
  68 #[test]
  69 fn test_counter_from_iter() {
  70     let it = (0..).step_by(5).take(10);
  71     let xs: Vec<isize> = FromIterator::from_iter(it);
  72     assert_eq!(xs, [0, 5, 10, 15, 20, 25, 30, 35, 40, 45]);
  73 }
  74
  75 #[test]
  76 fn test_iterator_chain() {
  77     let xs = [0, 1, 2, 3, 4, 5];
  78     let ys = [30, 40, 50, 60];
  79     let expected = [0, 1, 2, 3, 4, 5, 30, 40, 50, 60];
  80     let it = xs.iter().chain(&ys);
  81     let mut i = 0;
  82     for &x in it {
  83         assert_eq!(x, expected[i]);
  84         i += 1;
  85     }
  86     assert_eq!(i, expected.len());
  87
  88     let ys = (30..).step_by(10).take(4);
  89     let it = xs.iter().cloned().chain(ys);
  90     let mut i = 0;
  91     for x in it {
  92         assert_eq!(x, expected[i]);
  93         i += 1;
  94     }
  95     assert_eq!(i, expected.len());
  96 }
  97
  98 #[test]
  99 fn test_iterator_chain_nth() {
 100     let xs = [0, 1, 2, 3, 4, 5];
 101     let ys = [30, 40, 50, 60];
 102     let zs = [];
 103     let expected = [0, 1, 2, 3, 4, 5, 30, 40, 50, 60];
 104     for (i, x) in expected.iter().enumerate() {
 105         assert_eq!(Some(x), xs.iter().chain(&ys).nth(i));
 106     }
 107     assert_eq!(zs.iter().chain(&xs).nth(0), Some(&0));
 108
 109     let mut it = xs.iter().chain(&zs);
 110     assert_eq!(it.nth(5), Some(&5));
 111     assert_eq!(it.next(), None);
 112 }
 113
 114 #[test]
 115 fn test_iterator_chain_last() {
 116     let xs = [0, 1, 2, 3, 4, 5];
 117     let ys = [30, 40, 50, 60];
 118     let zs = [];
 119     assert_eq!(xs.iter().chain(&ys).last(), Some(&60));
 120     assert_eq!(zs.iter().chain(&ys).last(), Some(&60));
 121     assert_eq!(ys.iter().chain(&zs).last(), Some(&60));
 122     assert_eq!(zs.iter().chain(&zs).last(), None);
 123 }
 124
 125 #[test]
 126 fn test_iterator_chain_count() {
 127     let xs = [0, 1, 2, 3, 4, 5];
 128     let ys = [30, 40, 50, 60];
 129     let zs = [];
 130     assert_eq!(xs.iter().chain(&ys).count(), 10);
 131     assert_eq!(zs.iter().chain(&ys).count(), 4);
 132 }
 133
 134 #[test]
 135 fn test_iterator_chain_find() {
 136     let xs = [0, 1, 2, 3, 4, 5];
 137     let ys = [30, 40, 50, 60];
 138     let mut iter = xs.iter().chain(&ys);
 139     assert_eq!(iter.find(|&&i| i == 4), Some(&4));
 140     assert_eq!(iter.next(), Some(&5));
 141     assert_eq!(iter.find(|&&i| i == 40), Some(&40));
 142     assert_eq!(iter.next(), Some(&50));
 143     assert_eq!(iter.find(|&&i| i == 100), None);
 144     assert_eq!(iter.next(), None);
 145 }
 146
 147 #[test]
 148 fn test_filter_map() {
 149     let it = (0..).step_by(1).take(10)
 150         .filter_map(|x| if x % 2 == 0 { Some(x*x) } else { None });
 151     assert_eq!(it.collect::<Vec<usize>>(), [0*0, 2*2, 4*4, 6*6, 8*8]);
 152 }
 153
 154 #[test]
 155 fn test_iterator_enumerate() {
 156     let xs = [0, 1, 2, 3, 4, 5];
 157     let it = xs.iter().enumerate();
 158     for (i, &x) in it {
 159         assert_eq!(i, x);
 160     }
 161 }
 162
 163 #[test]
 164 fn test_iterator_enumerate_nth() {
 165     let xs = [0, 1, 2, 3, 4, 5];
 166     for (i, &x) in xs.iter().enumerate() {
 167         assert_eq!(i, x);
 168     }
 169
 170     let mut it = xs.iter().enumerate();
 171     while let Some((i, &x)) = it.nth(0) {
 172         assert_eq!(i, x);
 173     }
 174
 175     let mut it = xs.iter().enumerate();
 176     while let Some((i, &x)) = it.nth(1) {
 177         assert_eq!(i, x);
 178     }
 179
 180     let (i, &x) = xs.iter().enumerate().nth(3).unwrap();
 181     assert_eq!(i, x);
 182     assert_eq!(i, 3);
 183 }
 184
 185 #[test]
 186 fn test_iterator_enumerate_count() {
 187     let xs = [0, 1, 2, 3, 4, 5];
 188     assert_eq!(xs.iter().count(), 6);
 189 }
 190
 191 #[test]
 192 fn test_iterator_filter_count() {
 193     let xs = [0, 1, 2, 3, 4, 5, 6, 7, 8];
 194     assert_eq!(xs.iter().filter(|&&x| x % 2 == 0).count(), 5);
 195 }
 196
 197 #[test]
 198 fn test_iterator_peekable() {
 199     let xs = vec![0, 1, 2, 3, 4, 5];
 200     let mut it = xs.iter().cloned().peekable();
 201
 202     assert_eq!(it.len(), 6);
 203     assert_eq!(it.peek().unwrap(), &0);
 204     assert_eq!(it.len(), 6);
 205     assert_eq!(it.next().unwrap(), 0);
 206     assert_eq!(it.len(), 5);
 207     assert_eq!(it.next().unwrap(), 1);
 208     assert_eq!(it.len(), 4);
 209     assert_eq!(it.next().unwrap(), 2);
 210     assert_eq!(it.len(), 3);
 211     assert_eq!(it.peek().unwrap(), &3);
 212     assert_eq!(it.len(), 3);
 213     assert_eq!(it.peek().unwrap(), &3);
 214     assert_eq!(it.len(), 3);
 215     assert_eq!(it.next().unwrap(), 3);
 216     assert_eq!(it.len(), 2);
 217     assert_eq!(it.next().unwrap(), 4);
 218     assert_eq!(it.len(), 1);
 219     assert_eq!(it.peek().unwrap(), &5);
 220     assert_eq!(it.len(), 1);
 221     assert_eq!(it.next().unwrap(), 5);
 222     assert_eq!(it.len(), 0);
 223     assert!(it.peek().is_none());
 224     assert_eq!(it.len(), 0);
 225     assert!(it.next().is_none());
 226     assert_eq!(it.len(), 0);
 227 }
 228
 229 #[test]
 230 fn test_iterator_peekable_count() {
 231     let xs = [0, 1, 2, 3, 4, 5];
 232     let ys = [10];
 233     let zs: [i32; 0] = [];
 234
 235     assert_eq!(xs.iter().peekable().count(), 6);
 236
 237     let mut it = xs.iter().peekable();
 238     assert_eq!(it.peek(), Some(&&0));
 239     assert_eq!(it.count(), 6);
 240
 241     assert_eq!(ys.iter().peekable().count(), 1);
 242
 243     let mut it = ys.iter().peekable();
 244     assert_eq!(it.peek(), Some(&&10));
 245     assert_eq!(it.count(), 1);
 246
 247     assert_eq!(zs.iter().peekable().count(), 0);
 248
 249     let mut it = zs.iter().peekable();
 250     assert_eq!(it.peek(), None);
 251
 252 }
 253
 254 #[test]
 255 fn test_iterator_peekable_nth() {
 256     let xs = [0, 1, 2, 3, 4, 5];
 257     let mut it = xs.iter().peekable();
 258
 259     assert_eq!(it.peek(), Some(&&0));
 260     assert_eq!(it.nth(0), Some(&0));
 261     assert_eq!(it.peek(), Some(&&1));
 262     assert_eq!(it.nth(1), Some(&2));
 263     assert_eq!(it.peek(), Some(&&3));
 264     assert_eq!(it.nth(2), Some(&5));
 265     assert_eq!(it.next(), None);
 266 }
 267
 268 #[test]
 269 fn test_iterator_peekable_last() {
 270     let xs = [0, 1, 2, 3, 4, 5];
 271     let ys = [0];
 272
 273     let mut it = xs.iter().peekable();
 274     assert_eq!(it.peek(), Some(&&0));
 275     assert_eq!(it.last(), Some(&5));
 276
 277     let mut it = ys.iter().peekable();
 278     assert_eq!(it.peek(), Some(&&0));
 279     assert_eq!(it.last(), Some(&0));
 280
 281     let mut it = ys.iter().peekable();
 282     assert_eq!(it.next(), Some(&0));
 283     assert_eq!(it.peek(), None);
 284     assert_eq!(it.last(), None);
 285 }
 286
 287 /// This is an iterator that follows the Iterator contract,
 288 /// but it is not fused. After having returned None once, it will start
 289 /// producing elements if .next() is called again.
 290 pub struct CycleIter<'a, T: 'a> {
 291     index: usize,
 292     data: &'a [T],
 293 }
 294
 295 pub fn cycle<T>(data: &[T]) -> CycleIter<T> {
 296     CycleIter {
 297         index: 0,
 298         data: data,
 299     }
 300 }
 301
 302 impl<'a, T> Iterator for CycleIter<'a, T> {
 303     type Item = &'a T;
 304     fn next(&mut self) -> Option<Self::Item> {
 305         let elt = self.data.get(self.index);
 306         self.index += 1;
 307         self.index %= 1 + self.data.len();
 308         elt
 309     }
 310 }
 311
 312 #[test]
 313 fn test_iterator_peekable_remember_peek_none_1() {
 314     // Check that the loop using .peek() terminates
 315     let data = [1, 2, 3];
 316     let mut iter = cycle(&data).peekable();
 317
 318     let mut n = 0;
 319     while let Some(_) = iter.next() {
 320         let is_the_last = iter.peek().is_none();
 321         assert_eq!(is_the_last, n == data.len() - 1);
 322         n += 1;
 323         if n > data.len() { break; }
 324     }
 325     assert_eq!(n, data.len());
 326 }
 327
 328 #[test]
 329 fn test_iterator_peekable_remember_peek_none_2() {
 330     let data = [0];
 331     let mut iter = cycle(&data).peekable();
 332     iter.next();
 333     assert_eq!(iter.peek(), None);
 334     assert_eq!(iter.last(), None);
 335 }
 336
 337 #[test]
 338 fn test_iterator_peekable_remember_peek_none_3() {
 339     let data = [0];
 340     let mut iter = cycle(&data).peekable();
 341     iter.peek();
 342     assert_eq!(iter.nth(0), Some(&0));
 343
 344     let mut iter = cycle(&data).peekable();
 345     iter.next();
 346     assert_eq!(iter.peek(), None);
 347     assert_eq!(iter.nth(0), None);
 348 }
 349
 350 #[test]
 351 fn test_iterator_take_while() {
 352     let xs = [0, 1, 2, 3, 5, 13, 15, 16, 17, 19];
 353     let ys = [0, 1, 2, 3, 5, 13];
 354     let it = xs.iter().take_while(|&x| *x < 15);
 355     let mut i = 0;
 356     for x in it {
 357         assert_eq!(*x, ys[i]);
 358         i += 1;
 359     }
 360     assert_eq!(i, ys.len());
 361 }
 362
 363 #[test]
 364 fn test_iterator_skip_while() {
 365     let xs = [0, 1, 2, 3, 5, 13, 15, 16, 17, 19];
 366     let ys = [15, 16, 17, 19];
 367     let it = xs.iter().skip_while(|&x| *x < 15);
 368     let mut i = 0;
 369     for x in it {
 370         assert_eq!(*x, ys[i]);
 371         i += 1;
 372     }
 373     assert_eq!(i, ys.len());
 374 }
 375
 376 #[test]
 377 fn test_iterator_skip() {
 378     let xs = [0, 1, 2, 3, 5, 13, 15, 16, 17, 19, 20, 30];
 379     let ys = [13, 15, 16, 17, 19, 20, 30];
 380     let mut it = xs.iter().skip(5);
 381     let mut i = 0;
 382     while let Some(&x) = it.next() {
 383         assert_eq!(x, ys[i]);
 384         i += 1;
 385         assert_eq!(it.len(), xs.len()-5-i);
 386     }
 387     assert_eq!(i, ys.len());
 388     assert_eq!(it.len(), 0);
 389 }
 390
 391 #[test]
 392 fn test_iterator_skip_doubleended() {
 393     let xs = [0, 1, 2, 3, 5, 13, 15, 16, 17, 19, 20, 30];
 394     let mut it = xs.iter().rev().skip(5);
 395     assert_eq!(it.next(), Some(&15));
 396     assert_eq!(it.by_ref().rev().next(), Some(&0));
 397     assert_eq!(it.next(), Some(&13));
 398     assert_eq!(it.by_ref().rev().next(), Some(&1));
 399     assert_eq!(it.next(), Some(&5));
 400     assert_eq!(it.by_ref().rev().next(), Some(&2));
 401     assert_eq!(it.next(), Some(&3));
 402     assert_eq!(it.next(), None);
 403     let mut it = xs.iter().rev().skip(5).rev();
 404     assert_eq!(it.next(), Some(&0));
 405     assert_eq!(it.rev().next(), Some(&15));
 406     let mut it_base = xs.iter();
 407     {
 408         let mut it = it_base.by_ref().skip(5).rev();
 409         assert_eq!(it.next(), Some(&30));
 410         assert_eq!(it.next(), Some(&20));
 411         assert_eq!(it.next(), Some(&19));
 412         assert_eq!(it.next(), Some(&17));
 413         assert_eq!(it.next(), Some(&16));
 414         assert_eq!(it.next(), Some(&15));
 415         assert_eq!(it.next(), Some(&13));
 416         assert_eq!(it.next(), None);
 417     }
 418     // make sure the skipped parts have not been consumed
 419     assert_eq!(it_base.next(), Some(&0));
 420     assert_eq!(it_base.next(), Some(&1));
 421     assert_eq!(it_base.next(), Some(&2));
 422     assert_eq!(it_base.next(), Some(&3));
 423     assert_eq!(it_base.next(), Some(&5));
 424     assert_eq!(it_base.next(), None);
 425     let it = xs.iter().skip(5).rev();
 426     assert_eq!(it.last(), Some(&13));
 427 }
 428
 429 #[test]
 430 fn test_iterator_skip_nth() {
 431     let xs = [0, 1, 2, 3, 5, 13, 15, 16, 17, 19, 20, 30];
 432
 433     let mut it = xs.iter().skip(0);
 434     assert_eq!(it.nth(0), Some(&0));
 435     assert_eq!(it.nth(1), Some(&2));
 436
 437     let mut it = xs.iter().skip(5);
 438     assert_eq!(it.nth(0), Some(&13));
 439     assert_eq!(it.nth(1), Some(&16));
 440
 441     let mut it = xs.iter().skip(12);
 442     assert_eq!(it.nth(0), None);
 443
 444 }
 445
 446 #[test]
 447 fn test_iterator_skip_count() {
 448     let xs = [0, 1, 2, 3, 5, 13, 15, 16, 17, 19, 20, 30];
 449
 450     assert_eq!(xs.iter().skip(0).count(), 12);
 451     assert_eq!(xs.iter().skip(1).count(), 11);
 452     assert_eq!(xs.iter().skip(11).count(), 1);
 453     assert_eq!(xs.iter().skip(12).count(), 0);
 454     assert_eq!(xs.iter().skip(13).count(), 0);
 455 }
 456
 457 #[test]
 458 fn test_iterator_skip_last() {
 459     let xs = [0, 1, 2, 3, 5, 13, 15, 16, 17, 19, 20, 30];
 460
 461     assert_eq!(xs.iter().skip(0).last(), Some(&30));
 462     assert_eq!(xs.iter().skip(1).last(), Some(&30));
 463     assert_eq!(xs.iter().skip(11).last(), Some(&30));
 464     assert_eq!(xs.iter().skip(12).last(), None);
 465     assert_eq!(xs.iter().skip(13).last(), None);
 466
 467     let mut it = xs.iter().skip(5);
 468     assert_eq!(it.next(), Some(&13));
 469     assert_eq!(it.last(), Some(&30));
 470 }
 471
 472 #[test]
 473 fn test_iterator_take() {
 474     let xs = [0, 1, 2, 3, 5, 13, 15, 16, 17, 19];
 475     let ys = [0, 1, 2, 3, 5];
 476     let mut it = xs.iter().take(5);
 477     let mut i = 0;
 478     assert_eq!(it.len(), 5);
 479     while let Some(&x) = it.next() {
 480         assert_eq!(x, ys[i]);
 481         i += 1;
 482         assert_eq!(it.len(), 5-i);
 483     }
 484     assert_eq!(i, ys.len());
 485     assert_eq!(it.len(), 0);
 486 }
 487
 488 #[test]
 489 fn test_iterator_take_nth() {
 490     let xs = [0, 1, 2, 4, 5];
 491     let mut it = xs.iter();
 492     {
 493         let mut take = it.by_ref().take(3);
 494         let mut i = 0;
 495         while let Some(&x) = take.nth(0) {
 496             assert_eq!(x, i);
 497             i += 1;
 498         }
 499     }
 500     assert_eq!(it.nth(1), Some(&5));
 501     assert_eq!(it.nth(0), None);
 502
 503     let xs = [0, 1, 2, 3, 4];
 504     let mut it = xs.iter().take(7);
 505     let mut i = 1;
 506     while let Some(&x) = it.nth(1) {
 507         assert_eq!(x, i);
 508         i += 2;
 509     }
 510 }
 511
 512 #[test]
 513 fn test_iterator_take_short() {
 514     let xs = [0, 1, 2, 3];
 515     let ys = [0, 1, 2, 3];
 516     let mut it = xs.iter().take(5);
 517     let mut i = 0;
 518     assert_eq!(it.len(), 4);
 519     while let Some(&x) = it.next() {
 520         assert_eq!(x, ys[i]);
 521         i += 1;
 522         assert_eq!(it.len(), 4-i);
 523     }
 524     assert_eq!(i, ys.len());
 525     assert_eq!(it.len(), 0);
 526 }
 527
 528 #[test]
 529 fn test_iterator_scan() {
 530     // test the type inference
 531     fn add(old: &mut isize, new: &usize) -> Option<f64> {
 532         *old += *new as isize;
 533         Some(*old as f64)
 534     }
 535     let xs = [0, 1, 2, 3, 4];
 536     let ys = [0f64, 1.0, 3.0, 6.0, 10.0];
 537
 538     let it = xs.iter().scan(0, add);
 539     let mut i = 0;
 540     for x in it {
 541         assert_eq!(x, ys[i]);
 542         i += 1;
 543     }
 544     assert_eq!(i, ys.len());
 545 }
 546
 547 #[test]
 548 fn test_iterator_flat_map() {
 549     let xs = [0, 3, 6];
 550     let ys = [0, 1, 2, 3, 4, 5, 6, 7, 8];
 551     let it = xs.iter().flat_map(|&x| (x..).step_by(1).take(3));
 552     let mut i = 0;
 553     for x in it {
 554         assert_eq!(x, ys[i]);
 555         i += 1;
 556     }
 557     assert_eq!(i, ys.len());
 558 }
 559
 560 #[test]
 561 fn test_inspect() {
 562     let xs = [1, 2, 3, 4];
 563     let mut n = 0;
 564
 565     let ys = xs.iter()
 566                .cloned()
 567                .inspect(|_| n += 1)
 568                .collect::<Vec<usize>>();
 569
 570     assert_eq!(n, xs.len());
 571     assert_eq!(&xs[..], &ys[..]);
 572 }
 573
 574 #[test]
 575 fn test_cycle() {
 576     let cycle_len = 3;
 577     let it = (0..).step_by(1).take(cycle_len).cycle();
 578     assert_eq!(it.size_hint(), (usize::MAX, None));
 579     for (i, x) in it.take(100).enumerate() {
 580         assert_eq!(i % cycle_len, x);
 581     }
 582
 583     let mut it = (0..).step_by(1).take(0).cycle();
 584     assert_eq!(it.size_hint(), (0, Some(0)));
 585     assert_eq!(it.next(), None);
 586 }
 587
 588 #[test]
 589 fn test_iterator_nth() {
 590     let v: &[_] = &[0, 1, 2, 3, 4];
 591     for i in 0..v.len() {
 592         assert_eq!(v.iter().nth(i).unwrap(), &v[i]);
 593     }
 594     assert_eq!(v.iter().nth(v.len()), None);
 595 }
 596
 597 #[test]
 598 fn test_iterator_last() {
 599     let v: &[_] = &[0, 1, 2, 3, 4];
 600     assert_eq!(v.iter().last().unwrap(), &4);
 601     assert_eq!(v[..1].iter().last().unwrap(), &0);
 602 }
 603
 604 #[test]
 605 fn test_iterator_len() {
 606     let v: &[_] = &[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10];
 607     assert_eq!(v[..4].iter().count(), 4);
 608     assert_eq!(v[..10].iter().count(), 10);
 609     assert_eq!(v[..0].iter().count(), 0);
 610 }
 611
 612 #[test]
 613 fn test_iterator_sum() {
 614     let v: &[i32] = &[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10];
 615     assert_eq!(v[..4].iter().cloned().sum::<i32>(), 6);
 616     assert_eq!(v.iter().cloned().sum::<i32>(), 55);
 617     assert_eq!(v[..0].iter().cloned().sum::<i32>(), 0);
 618 }
 619
 620 #[test]
 621 fn test_iterator_sum_result() {
 622     let v: &[Result<i32, ()>] = &[Ok(1), Ok(2), Ok(3), Ok(4)];
 623     assert_eq!(v.iter().cloned().sum::<Result<i32, _>>(), Ok(10));
 624     let v: &[Result<i32, ()>] = &[Ok(1), Err(()), Ok(3), Ok(4)];
 625     assert_eq!(v.iter().cloned().sum::<Result<i32, _>>(), Err(()));
 626 }
 627
 628 #[test]
 629 fn test_iterator_product() {
 630     let v: &[i32] = &[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10];
 631     assert_eq!(v[..4].iter().cloned().product::<i32>(), 0);
 632     assert_eq!(v[1..5].iter().cloned().product::<i32>(), 24);
 633     assert_eq!(v[..0].iter().cloned().product::<i32>(), 1);
 634 }
 635
 636 #[test]
 637 fn test_iterator_product_result() {
 638     let v: &[Result<i32, ()>] = &[Ok(1), Ok(2), Ok(3), Ok(4)];
 639     assert_eq!(v.iter().cloned().product::<Result<i32, _>>(), Ok(24));
 640     let v: &[Result<i32, ()>] = &[Ok(1), Err(()), Ok(3), Ok(4)];
 641     assert_eq!(v.iter().cloned().product::<Result<i32, _>>(), Err(()));
 642 }
 643
 644 #[test]
 645 fn test_iterator_max() {
 646     let v: &[_] = &[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10];
 647     assert_eq!(v[..4].iter().cloned().max(), Some(3));
 648     assert_eq!(v.iter().cloned().max(), Some(10));
 649     assert_eq!(v[..0].iter().cloned().max(), None);
 650 }
 651
 652 #[test]
 653 fn test_iterator_min() {
 654     let v: &[_] = &[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10];
 655     assert_eq!(v[..4].iter().cloned().min(), Some(0));
 656     assert_eq!(v.iter().cloned().min(), Some(0));
 657     assert_eq!(v[..0].iter().cloned().min(), None);
 658 }
 659
 660 #[test]
 661 fn test_iterator_size_hint() {
 662     let c = (0..).step_by(1);
 663     let v: &[_] = &[0, 1, 2, 3, 4, 5, 6, 7, 8, 9];
 664     let v2 = &[10, 11, 12];
 665     let vi = v.iter();
 666
 667     assert_eq!(c.size_hint(), (usize::MAX, None));
 668     assert_eq!(vi.clone().size_hint(), (10, Some(10)));
 669
 670     assert_eq!(c.clone().take(5).size_hint(), (5, Some(5)));
 671     assert_eq!(c.clone().skip(5).size_hint().1, None);
 672     assert_eq!(c.clone().take_while(|_| false).size_hint(), (0, None));
 673     assert_eq!(c.clone().skip_while(|_| false).size_hint(), (0, None));
 674     assert_eq!(c.clone().enumerate().size_hint(), (usize::MAX, None));
 675     assert_eq!(c.clone().chain(vi.clone().cloned()).size_hint(), (usize::MAX, None));
 676     assert_eq!(c.clone().zip(vi.clone()).size_hint(), (10, Some(10)));
 677     assert_eq!(c.clone().scan(0, |_,_| Some(0)).size_hint(), (0, None));
 678     assert_eq!(c.clone().filter(|_| false).size_hint(), (0, None));
 679     assert_eq!(c.clone().map(|_| 0).size_hint(), (usize::MAX, None));
 680     assert_eq!(c.filter_map(|_| Some(0)).size_hint(), (0, None));
 681
 682     assert_eq!(vi.clone().take(5).size_hint(), (5, Some(5)));
 683     assert_eq!(vi.clone().take(12).size_hint(), (10, Some(10)));
 684     assert_eq!(vi.clone().skip(3).size_hint(), (7, Some(7)));
 685     assert_eq!(vi.clone().skip(12).size_hint(), (0, Some(0)));
 686     assert_eq!(vi.clone().take_while(|_| false).size_hint(), (0, Some(10)));
 687     assert_eq!(vi.clone().skip_while(|_| false).size_hint(), (0, Some(10)));
 688     assert_eq!(vi.clone().enumerate().size_hint(), (10, Some(10)));
 689     assert_eq!(vi.clone().chain(v2).size_hint(), (13, Some(13)));
 690     assert_eq!(vi.clone().zip(v2).size_hint(), (3, Some(3)));
 691     assert_eq!(vi.clone().scan(0, |_,_| Some(0)).size_hint(), (0, Some(10)));
 692     assert_eq!(vi.clone().filter(|_| false).size_hint(), (0, Some(10)));
 693     assert_eq!(vi.clone().map(|&i| i+1).size_hint(), (10, Some(10)));
 694     assert_eq!(vi.filter_map(|_| Some(0)).size_hint(), (0, Some(10)));
 695 }
 696
 697 #[test]
 698 fn test_collect() {
 699     let a = vec![1, 2, 3, 4, 5];
 700     let b: Vec<isize> = a.iter().cloned().collect();
 701     assert!(a == b);
 702 }
 703
 704 #[test]
 705 fn test_all() {
 706     let v: Box<[isize]> = Box::new([1, 2, 3, 4, 5]);
 707     assert!(v.iter().all(|&x| x < 10));
 708     assert!(!v.iter().all(|&x| x % 2 == 0));
 709     assert!(!v.iter().all(|&x| x > 100));
 710     assert!(v[..0].iter().all(|_| panic!()));
 711 }
 712
 713 #[test]
 714 fn test_any() {
 715     let v: Box<[isize]> = Box::new([1, 2, 3, 4, 5]);
 716     assert!(v.iter().any(|&x| x < 10));
 717     assert!(v.iter().any(|&x| x % 2 == 0));
 718     assert!(!v.iter().any(|&x| x > 100));
 719     assert!(!v[..0].iter().any(|_| panic!()));
 720 }
 721
 722 #[test]
 723 fn test_find() {
 724     let v: &[isize] = &[1, 3, 9, 27, 103, 14, 11];
 725     assert_eq!(*v.iter().find(|&&x| x & 1 == 0).unwrap(), 14);
 726     assert_eq!(*v.iter().find(|&&x| x % 3 == 0).unwrap(), 3);
 727     assert!(v.iter().find(|&&x| x % 12 == 0).is_none());
 728 }
 729
 730 #[test]
 731 fn test_position() {
 732     let v = &[1, 3, 9, 27, 103, 14, 11];
 733     assert_eq!(v.iter().position(|x| *x & 1 == 0).unwrap(), 5);
 734     assert_eq!(v.iter().position(|x| *x % 3 == 0).unwrap(), 1);
 735     assert!(v.iter().position(|x| *x % 12 == 0).is_none());
 736 }
 737
 738 #[test]
 739 fn test_count() {
 740     let xs = &[1, 2, 2, 1, 5, 9, 0, 2];
 741     assert_eq!(xs.iter().filter(|x| **x == 2).count(), 3);
 742     assert_eq!(xs.iter().filter(|x| **x == 5).count(), 1);
 743     assert_eq!(xs.iter().filter(|x| **x == 95).count(), 0);
 744 }
 745
 746 #[test]
 747 fn test_max_by_key() {
 748     let xs: &[isize] = &[-3, 0, 1, 5, -10];
 749     assert_eq!(*xs.iter().max_by_key(|x| x.abs()).unwrap(), -10);
 750 }
 751
 752 #[test]
 753 fn test_max_by() {
 754     let xs: &[isize] = &[-3, 0, 1, 5, -10];
 755     assert_eq!(*xs.iter().max_by(|x, y| x.abs().cmp(&y.abs())).unwrap(), -10);
 756 }
 757
 758 #[test]
 759 fn test_min_by_key() {
 760     let xs: &[isize] = &[-3, 0, 1, 5, -10];
 761     assert_eq!(*xs.iter().min_by_key(|x| x.abs()).unwrap(), 0);
 762 }
 763
 764 #[test]
 765 fn test_min_by() {
 766     let xs: &[isize] = &[-3, 0, 1, 5, -10];
 767     assert_eq!(*xs.iter().min_by(|x, y| x.abs().cmp(&y.abs())).unwrap(), 0);
 768 }
 769
 770 #[test]
 771 fn test_by_ref() {
 772     let mut xs = 0..10;
 773     // sum the first five values
 774     let partial_sum = xs.by_ref().take(5).fold(0, |a, b| a + b);
 775     assert_eq!(partial_sum, 10);
 776     assert_eq!(xs.next(), Some(5));
 777 }
 778
 779 #[test]
 780 fn test_rev() {
 781     let xs = [2, 4, 6, 8, 10, 12, 14, 16];
 782     let mut it = xs.iter();
 783     it.next();
 784     it.next();
 785     assert!(it.rev().cloned().collect::<Vec<isize>>() ==
 786             vec![16, 14, 12, 10, 8, 6]);
 787 }
 788
 789 #[test]
 790 fn test_cloned() {
 791     let xs = [2, 4, 6, 8];
 792
 793     let mut it = xs.iter().cloned();
 794     assert_eq!(it.len(), 4);
 795     assert_eq!(it.next(), Some(2));
 796     assert_eq!(it.len(), 3);
 797     assert_eq!(it.next(), Some(4));
 798     assert_eq!(it.len(), 2);
 799     assert_eq!(it.next_back(), Some(8));
 800     assert_eq!(it.len(), 1);
 801     assert_eq!(it.next_back(), Some(6));
 802     assert_eq!(it.len(), 0);
 803     assert_eq!(it.next_back(), None);
 804 }
 805
 806 #[test]
 807 fn test_double_ended_map() {
 808     let xs = [1, 2, 3, 4, 5, 6];
 809     let mut it = xs.iter().map(|&x| x * -1);
 810     assert_eq!(it.next(), Some(-1));
 811     assert_eq!(it.next(), Some(-2));
 812     assert_eq!(it.next_back(), Some(-6));
 813     assert_eq!(it.next_back(), Some(-5));
 814     assert_eq!(it.next(), Some(-3));
 815     assert_eq!(it.next_back(), Some(-4));
 816     assert_eq!(it.next(), None);
 817 }
 818
 819 #[test]
 820 fn test_double_ended_enumerate() {
 821     let xs = [1, 2, 3, 4, 5, 6];
 822     let mut it = xs.iter().cloned().enumerate();
 823     assert_eq!(it.next(), Some((0, 1)));
 824     assert_eq!(it.next(), Some((1, 2)));
 825     assert_eq!(it.next_back(), Some((5, 6)));
 826     assert_eq!(it.next_back(), Some((4, 5)));
 827     assert_eq!(it.next_back(), Some((3, 4)));
 828     assert_eq!(it.next_back(), Some((2, 3)));
 829     assert_eq!(it.next(), None);
 830 }
 831
 832 #[test]
 833 fn test_double_ended_zip() {
 834     let xs = [1, 2, 3, 4, 5, 6];
 835     let ys = [1, 2, 3, 7];
 836     let a = xs.iter().cloned();
 837     let b = ys.iter().cloned();
 838     let mut it = a.zip(b);
 839     assert_eq!(it.next(), Some((1, 1)));
 840     assert_eq!(it.next(), Some((2, 2)));
 841     assert_eq!(it.next_back(), Some((4, 7)));
 842     assert_eq!(it.next_back(), Some((3, 3)));
 843     assert_eq!(it.next(), None);
 844 }
 845
 846 #[test]
 847 fn test_double_ended_filter() {
 848     let xs = [1, 2, 3, 4, 5, 6];
 849     let mut it = xs.iter().filter(|&x| *x & 1 == 0);
 850     assert_eq!(it.next_back().unwrap(), &6);
 851     assert_eq!(it.next_back().unwrap(), &4);
 852     assert_eq!(it.next().unwrap(), &2);
 853     assert_eq!(it.next_back(), None);
 854 }
 855
 856 #[test]
 857 fn test_double_ended_filter_map() {
 858     let xs = [1, 2, 3, 4, 5, 6];
 859     let mut it = xs.iter().filter_map(|&x| if x & 1 == 0 { Some(x * 2) } else { None });
 860     assert_eq!(it.next_back().unwrap(), 12);
 861     assert_eq!(it.next_back().unwrap(), 8);
 862     assert_eq!(it.next().unwrap(), 4);
 863     assert_eq!(it.next_back(), None);
 864 }
 865
 866 #[test]
 867 fn test_double_ended_chain() {
 868     let xs = [1, 2, 3, 4, 5];
 869     let ys = [7, 9, 11];
 870     let mut it = xs.iter().chain(&ys).rev();
 871     assert_eq!(it.next().unwrap(), &11);
 872     assert_eq!(it.next().unwrap(), &9);
 873     assert_eq!(it.next_back().unwrap(), &1);
 874     assert_eq!(it.next_back().unwrap(), &2);
 875     assert_eq!(it.next_back().unwrap(), &3);
 876     assert_eq!(it.next_back().unwrap(), &4);
 877     assert_eq!(it.next_back().unwrap(), &5);
 878     assert_eq!(it.next_back().unwrap(), &7);
 879     assert_eq!(it.next_back(), None);
 880
 881
 882     // test that .chain() is well behaved with an unfused iterator
 883     struct CrazyIterator(bool);
 884     impl CrazyIterator { fn new() -> CrazyIterator { CrazyIterator(false) } }
 885     impl Iterator for CrazyIterator {
 886         type Item = i32;
 887         fn next(&mut self) -> Option<i32> {
 888             if self.0 { Some(99) } else { self.0 = true; None }
 889         }
 890     }
 891
 892     impl DoubleEndedIterator for CrazyIterator {
 893         fn next_back(&mut self) -> Option<i32> {
 894             self.next()
 895         }
 896     }
 897
 898     assert_eq!(CrazyIterator::new().chain(0..10).rev().last(), Some(0));
 899     assert!((0..10).chain(CrazyIterator::new()).rev().any(|i| i == 0));
 900 }
 901
 902 #[test]
 903 fn test_rposition() {
 904     fn f(xy: &(isize, char)) -> bool { let (_x, y) = *xy; y == 'b' }
 905     fn g(xy: &(isize, char)) -> bool { let (_x, y) = *xy; y == 'd' }
 906     let v = [(0, 'a'), (1, 'b'), (2, 'c'), (3, 'b')];
 907
 908     assert_eq!(v.iter().rposition(f), Some(3));
 909     assert!(v.iter().rposition(g).is_none());
 910 }
 911
 912 #[test]
 913 #[should_panic]
 914 fn test_rposition_panic() {
 915     let v: [(Box<_>, Box<_>); 4] =
 916         [(box 0, box 0), (box 0, box 0),
 917          (box 0, box 0), (box 0, box 0)];
 918     let mut i = 0;
 919     v.iter().rposition(|_elt| {
 920         if i == 2 {
 921             panic!()
 922         }
 923         i += 1;
 924         false
 925     });
 926 }
 927
 928
 929 #[test]
 930 fn test_double_ended_flat_map() {
 931     let u = [0,1];
 932     let v = [5,6,7,8];
 933     let mut it = u.iter().flat_map(|x| &v[*x..v.len()]);
 934     assert_eq!(it.next_back().unwrap(), &8);
 935     assert_eq!(it.next().unwrap(),      &5);
 936     assert_eq!(it.next_back().unwrap(), &7);
 937     assert_eq!(it.next_back().unwrap(), &6);
 938     assert_eq!(it.next_back().unwrap(), &8);
 939     assert_eq!(it.next().unwrap(),      &6);
 940     assert_eq!(it.next_back().unwrap(), &7);
 941     assert_eq!(it.next_back(), None);
 942     assert_eq!(it.next(),      None);
 943     assert_eq!(it.next_back(), None);
 944 }
 945
 946 #[test]
 947 fn test_double_ended_range() {
 948     assert_eq!((11..14).rev().collect::<Vec<_>>(), [13, 12, 11]);
 949     for _ in (10..0).rev() {
 950         panic!("unreachable");
 951     }
 952
 953     assert_eq!((11..14).rev().collect::<Vec<_>>(), [13, 12, 11]);
 954     for _ in (10..0).rev() {
 955         panic!("unreachable");
 956     }
 957 }
 958
 959 #[test]
 960 fn test_range() {
 961     assert_eq!((0..5).collect::<Vec<_>>(), [0, 1, 2, 3, 4]);
 962     assert_eq!((-10..-1).collect::<Vec<_>>(), [-10, -9, -8, -7, -6, -5, -4, -3, -2]);
 963     assert_eq!((0..5).rev().collect::<Vec<_>>(), [4, 3, 2, 1, 0]);
 964     assert_eq!((200..-5).count(), 0);
 965     assert_eq!((200..-5).rev().count(), 0);
 966     assert_eq!((200..200).count(), 0);
 967     assert_eq!((200..200).rev().count(), 0);
 968
 969     assert_eq!((0..100).size_hint(), (100, Some(100)));
 970     // this test is only meaningful when sizeof usize < sizeof u64
 971     assert_eq!((usize::MAX - 1..usize::MAX).size_hint(), (1, Some(1)));
 972     assert_eq!((-10..-1).size_hint(), (9, Some(9)));
 973     assert_eq!((-1..-10).size_hint(), (0, Some(0)));
 974
 975     assert_eq!((-70..58).size_hint(), (128, Some(128)));
 976     assert_eq!((-128..127).size_hint(), (255, Some(255)));
 977     assert_eq!((-2..isize::MAX).size_hint(),
 978                (isize::MAX as usize + 2, Some(isize::MAX as usize + 2)));
 979 }
 980
 981 #[test]
 982 fn test_range_step() {
 983     assert_eq!((0..20).step_by(5).collect::<Vec<isize>>(), [0, 5, 10, 15]);
 984     assert_eq!((20..0).step_by(-5).collect::<Vec<isize>>(), [20, 15, 10, 5]);
 985     assert_eq!((20..0).step_by(-6).collect::<Vec<isize>>(), [20, 14, 8, 2]);
 986     assert_eq!((200..255).step_by(50).collect::<Vec<u8>>(), [200, 250]);
 987     assert_eq!((200..-5).step_by(1).collect::<Vec<isize>>(), []);
 988     assert_eq!((200..200).step_by(1).collect::<Vec<isize>>(), []);
 989
 990     assert_eq!((0..20).step_by(1).size_hint(), (20, Some(20)));
 991     assert_eq!((0..20).step_by(21).size_hint(), (1, Some(1)));
 992     assert_eq!((0..20).step_by(5).size_hint(), (4, Some(4)));
 993     assert_eq!((20..0).step_by(-5).size_hint(), (4, Some(4)));
 994     assert_eq!((20..0).step_by(-6).size_hint(), (4, Some(4)));
 995     assert_eq!((20..-5).step_by(1).size_hint(), (0, Some(0)));
 996     assert_eq!((20..20).step_by(1).size_hint(), (0, Some(0)));
 997     assert_eq!((0..1).step_by(0).size_hint(), (0, None));
 998     assert_eq!((i8::MAX..i8::MIN).step_by(i8::MIN).size_hint(), (2, Some(2)));
 999     assert_eq!((i16::MIN..i16::MAX).step_by(i16::MAX).size_hint(), (3, Some(3)));
1000     assert_eq!((isize::MIN..isize::MAX).step_by(1).size_hint(), (usize::MAX, Some(usize::MAX)));
1001 }
1002
1003 #[test]
1004 fn test_repeat() {
1005     let mut it = repeat(42);
1006     assert_eq!(it.next(), Some(42));
1007     assert_eq!(it.next(), Some(42));
1008     assert_eq!(it.next(), Some(42));
1009 }
1010
1011 #[test]
1012 fn test_fuse() {
1013     let mut it = 0..3;
1014     assert_eq!(it.len(), 3);
1015     assert_eq!(it.next(), Some(0));
1016     assert_eq!(it.len(), 2);
1017     assert_eq!(it.next(), Some(1));
1018     assert_eq!(it.len(), 1);
1019     assert_eq!(it.next(), Some(2));
1020     assert_eq!(it.len(), 0);
1021     assert_eq!(it.next(), None);
1022     assert_eq!(it.len(), 0);
1023     assert_eq!(it.next(), None);
1024     assert_eq!(it.len(), 0);
1025     assert_eq!(it.next(), None);
1026     assert_eq!(it.len(), 0);
1027 }
1028
1029 #[test]
1030 fn test_fuse_nth() {
1031     let xs = [0, 1, 2];
1032     let mut it = xs.iter();
1033
1034     assert_eq!(it.len(), 3);
1035     assert_eq!(it.nth(2), Some(&2));
1036     assert_eq!(it.len(), 0);
1037     assert_eq!(it.nth(2), None);
1038     assert_eq!(it.len(), 0);
1039 }
1040
1041 #[test]
1042 fn test_fuse_last() {
1043     let xs = [0, 1, 2];
1044     let it = xs.iter();
1045
1046     assert_eq!(it.len(), 3);
1047     assert_eq!(it.last(), Some(&2));
1048 }
1049
1050 #[test]
1051 fn test_fuse_count() {
1052     let xs = [0, 1, 2];
1053     let it = xs.iter();
1054
1055     assert_eq!(it.len(), 3);
1056     assert_eq!(it.count(), 3);
1057     // Can't check len now because count consumes.
1058 }
1059
1060 #[test]
1061 fn test_once() {
1062     let mut it = once(42);
1063     assert_eq!(it.next(), Some(42));
1064     assert_eq!(it.next(), None);
1065 }
1066
1067 #[test]
1068 fn test_empty() {
1069     let mut it = empty::<i32>();
1070     assert_eq!(it.next(), None);
1071 }
1072
1073 #[test]
1074 fn test_chain_fold() {
1075     let xs = [1, 2, 3];
1076     let ys = [1, 2, 0];
1077
1078     let mut iter = xs.iter().chain(&ys);
1079     iter.next();
1080     let mut result = Vec::new();
1081     iter.fold((), |(), &elt| result.push(elt));
1082     assert_eq!(&[2, 3, 1, 2, 0], &result[..]);
1083 }
1084
1085 #[test]
1086 fn test_step_replace_unsigned() {
1087     let mut x = 4u32;
1088     let y = x.replace_zero();
1089     assert_eq!(x, 0);
1090     assert_eq!(y, 4);
1091
1092     x = 5;
1093     let y = x.replace_one();
1094     assert_eq!(x, 1);
1095     assert_eq!(y, 5);
1096 }
1097
1098 #[test]
1099 fn test_step_replace_signed() {
1100     let mut x = 4i32;
1101     let y = x.replace_zero();
1102     assert_eq!(x, 0);
1103     assert_eq!(y, 4);
1104
1105     x = 5;
1106     let y = x.replace_one();
1107     assert_eq!(x, 1);
1108     assert_eq!(y, 5);
1109 }
1110
1111 #[test]
1112 fn test_step_replace_no_between() {
1113     let mut x = 4u128;
1114     let y = x.replace_zero();
1115     assert_eq!(x, 0);
1116     assert_eq!(y, 4);
1117
1118     x = 5;
1119     let y = x.replace_one();
1120     assert_eq!(x, 1);
1121     assert_eq!(y, 5);
1122 }