diff options
Diffstat (limited to 'src/libstd')
| -rw-r--r-- | src/libstd/vec.rs | 17 |
1 files changed, 16 insertions, 1 deletions
diff --git a/src/libstd/vec.rs b/src/libstd/vec.rs index fb9c47b4373..1014ff48b1d 100644 --- a/src/libstd/vec.rs +++ b/src/libstd/vec.rs @@ -2025,7 +2025,12 @@ macro_rules! iterator { #[inline] fn size_hint(&self) -> (Option<uint>, Option<uint>) { - let exact = Some(((self.end as uint) - (self.ptr as uint)) / size_of::<$elem>()); + let diff = if $step > 0 { + (self.end as uint) - (self.ptr as uint) + } else { + (self.ptr as uint) - (self.end as uint) + }; + let exact = Some(diff / size_of::<$elem>()); (exact, exact) } } @@ -2978,6 +2983,16 @@ mod tests { } #[test] + fn test_iter_size_hints() { + use iterator::*; + let mut xs = [1, 2, 5, 10, 11]; + assert_eq!(xs.iter().size_hint(), (Some(5), Some(5))); + assert_eq!(xs.rev_iter().size_hint(), (Some(5), Some(5))); + assert_eq!(xs.mut_iter().size_hint(), (Some(5), Some(5))); + assert_eq!(xs.mut_rev_iter().size_hint(), (Some(5), Some(5))); + } + + #[test] fn test_mut_iterator() { use iterator::*; let mut xs = [1, 2, 3, 4, 5]; |
