From 0b4d8d688240c229a4e3d51cdaa96899258a2340 Mon Sep 17 00:00:00 2001
From: Steven Fackler <sfackler@gmail.com>
Date: Fri, 12 Jul 2013 02:13:26 -0400
Subject: [PATCH] Added iterator tests

---
 src/libextra/bitv.rs | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/src/libextra/bitv.rs b/src/libextra/bitv.rs
index f099593e86f..69c851e248b 100644
--- a/src/libextra/bitv.rs
+++ b/src/libextra/bitv.rs
@@ -1265,6 +1265,25 @@ mod tests {
         assert_eq!(from_bytes([0b00100110]).to_bools(), bools);
     }
 
+    #[test]
+    fn test_bitv_iterator() {
+        let bools = [true, false, true, true];
+        let bitv = from_bools(bools);
+
+        for bitv.iter().zip(bools.iter()).advance |(act, &ex)| {
+            assert_eq!(ex, act);
+        }
+    }
+
+    #[test]
+    fn test_bitv_set_iterator() {
+        let bools = [true, false, true, true];
+        let bitv = BitvSet::from_bitv(from_bools(bools));
+
+        let idxs: ~[uint] = bitv.iter().collect();
+        assert_eq!(idxs, ~[0, 2, 3]);
+    }
+
     #[test]
     fn test_small_difference() {
         let mut b1 = Bitv::new(3, false);