From 4ab7b213de8d8f8307f0a1877cb4a4d4c802e792 Mon Sep 17 00:00:00 2001
From: Wouter Geraedts <git@woutergeraedts.nl>
Date: Tue, 12 Nov 2019 15:38:02 +0100
Subject: [PATCH] Updated example to be consistent; added timing measurements
 to throttle

---
 examples/throttle.rs     |  2 +-
 src/stream/stream/mod.rs | 23 ++++++++++++++++++-----
 2 files changed, 19 insertions(+), 6 deletions(-)

diff --git a/examples/throttle.rs b/examples/throttle.rs
index 1b9a6f2..74c1fd3 100644
--- a/examples/throttle.rs
+++ b/examples/throttle.rs
@@ -11,7 +11,7 @@ fn main() {
             use std::time::Duration;
 
             // emit value every 1 second
-            let s = stream::interval(Duration::from_nanos(1000000)).enumerate();
+            let s = stream::interval(Duration::from_secs(1)).enumerate();
 
             // throttle for 2 seconds
             let s = s.throttle(Duration::from_secs(2));
diff --git a/src/stream/stream/mod.rs b/src/stream/stream/mod.rs
index 48a1a20..cec874f 100644
--- a/src/stream/stream/mod.rs
+++ b/src/stream/stream/mod.rs
@@ -325,19 +325,32 @@ extension_trait! {
             #
             use async_std::prelude::*;
             use async_std::stream;
-            use std::time::Duration;
+            use std::time::{Duration, Instant};
 
-            // emit value every 1 second
-            let s = stream::interval(Duration::from_millis(5)).enumerate().take(3);
+            // emit value every 5 milliseconds
+            let s = stream::interval(Duration::from_millis(5))
+                .enumerate()
+                .take(3);
 
-            // throttle for 2 seconds
+            // throttle for 10 milliseconds
             let mut s = s.throttle(Duration::from_millis(10));
 
+            let start = Instant::now();
             assert_eq!(s.next().await, Some((0, ())));
+            let duration_ms = start.elapsed().as_millis();
+            assert!(duration_ms >= 5 && duration_ms < 15);
+
             assert_eq!(s.next().await, Some((1, ())));
+            let duration_ms = start.elapsed().as_millis();
+            assert!(duration_ms >= 15 && duration_ms < 25);
+
             assert_eq!(s.next().await, Some((2, ())));
+            let duration_ms = start.elapsed().as_millis();
+            assert!(duration_ms >= 25 && duration_ms < 35);
+
             assert_eq!(s.next().await, None);
-            // with a pause of 2 seconds between each print
+            let duration_ms = start.elapsed().as_millis();
+            assert!(duration_ms >= 35 && duration_ms < 45);
             #
             # }) }
             ```