1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
// Copyright © 2024 Mikhail Hogrefe
//
// This file is part of Malachite.
//
// Malachite is free software: you can redistribute it and/or modify it under the terms of the GNU
// Lesser General Public License (LGPL) as published by the Free Software Foundation; either version
// 3 of the License, or (at your option) any later version. See <https://www.gnu.org/licenses/>.
use itertools::__std_iter::empty;
use malachite_base::iterators::iterator_cache::IteratorCache;
use malachite_base::nevers::Never;
use malachite_base::num::exhaustive::exhaustive_signeds;
#[test]
fn test_iterator_cache() {
let mut xs = IteratorCache::new(empty::<Never>());
assert_eq!(xs.known_len(), None);
assert_eq!(xs.get(1), None);
assert_eq!(xs.known_len(), Some(0));
assert_eq!(xs.get(0), None);
let mut xs = IteratorCache::new([1, 2, 3].iter().copied());
assert_eq!(xs.known_len(), None);
assert_eq!(xs.get(1), Some(&2));
assert_eq!(xs.assert_get(1), &2);
assert_eq!(xs.known_len(), None);
assert_eq!(xs.get(0), Some(&1));
assert_eq!(xs.assert_get(0), &1);
assert_eq!(xs.get(3), None);
assert_eq!(xs.known_len(), Some(3));
assert_eq!(xs.get(2), Some(&3));
assert_eq!(xs.assert_get(2), &3);
let mut xs = IteratorCache::new(exhaustive_signeds::<i64>());
assert_eq!(xs.get(1), Some(&1));
assert_eq!(xs.assert_get(1), &1);
assert_eq!(xs.known_len(), None);
assert_eq!(xs.get(0), Some(&0));
assert_eq!(xs.assert_get(0), &0);
assert_eq!(xs.get(3), Some(&2));
assert_eq!(xs.assert_get(3), &2);
assert_eq!(xs.get(100), Some(&-50));
assert_eq!(xs.assert_get(100), &-50);
assert_eq!(xs.known_len(), None);
}
#[test]
#[should_panic]
fn iterator_cache_fail() {
IteratorCache::new(empty::<Never>()).assert_get(0);
}
|
