Hour produce an hourly range

alphagov · Oct 18, 2013 · 568485b · 568485b
1 parent 71bfe39
commit 568485b
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/backdrop/core/timeseries.py b/backdrop/core/timeseries.py
@@ -38,6 +38,9 @@ def __init__(self):
         self.name = "hour"
         self._delta = timedelta(hours=1)
 
+    def _is_boundary(self, timestamp):
+        return self.valid_start_at(timestamp)
+
     def start(self, timestamp):
         return timestamp.replace(minute=0, second=0, microsecond=0)
 

diff --git a/tests/core/test_timeseries.py b/tests/core/test_timeseries.py
@@ -281,3 +281,13 @@ def test_that_returns_the_end_of_the_current_hour(self):
 
         assert_that(end, is_(d(2013, 10, 4, 11, 0, 0)))
 
+    def test_that_a_range_of_five_hours_gives_us_five_data_points(self):
+        range = HOUR.range(d_tz(2013, 4, 3, 12), d_tz(2013, 4, 3, 17))
+
+        assert_that(list(range), contains(
+            (d_tz(2013, 4, 3, 12), d_tz(2013, 4, 3, 13)),
+            (d_tz(2013, 4, 3, 13), d_tz(2013, 4, 3, 14)),
+            (d_tz(2013, 4, 3, 14), d_tz(2013, 4, 3, 15)),
+            (d_tz(2013, 4, 3, 15), d_tz(2013, 4, 3, 16)),
+            (d_tz(2013, 4, 3, 16), d_tz(2013, 4, 3, 17))
+        ))