Fixes range() behavior.

Author: sdispater

pendulum/period.py

@@ -98,10 +98,14 @@ def xrange(self, unit):
             op = operator.ge
 
         start, end = self.start, self.end
+
+        i = 1
         while op(start, end):
             yield start
 
-            start = getattr(start, method)(**{unit: 1})
+            start = getattr(self.start, method)(**{unit: i})
+
+            i += 1
 
     def intersect(self, *periods):
         """

tests/period_tests/test_range.py

@@ -73,3 +73,22 @@ def test_contains_with_datetime(self):
         p = Period(dt1, dt2)
         dt = datetime(2000, 1, 7)
         self.assertTrue(dt in p)
+
+    def test_range_months_overflow(self):
+        dt1 = Pendulum(2016, 1, 30, tzinfo='America/Sao_Paulo')
+        dt2 = dt1.add(months=4)
+
+        p = Period(dt1, dt2)
+        r = p.range('months')
+        self.assertPendulum(r[0], 2016, 1, 30, 0, 0, 0)
+        self.assertPendulum(r[-1], 2016, 5, 30, 0, 0, 0)
+
+    def test_range_with_dst(self):
+        dt1 = Pendulum(2016, 10, 14, tzinfo='America/Sao_Paulo')
+        dt2 = dt1.add(weeks=1)
+
+        p = Period(dt1, dt2)
+        r = p.range('days')
+        self.assertPendulum(r[0], 2016, 10, 14, 0, 0, 0)
+        self.assertPendulum(r[2], 2016, 10, 16, 1, 0, 0)
+        self.assertPendulum(r[-1], 2016, 10, 21, 0, 0, 0)