Fixes tests

Author: sdispater

tests/date_tests/test_diff.py

@@ -1,7 +1,7 @@
 # -*- coding: utf-8 -*-
 
 from datetime import date
-from pendulum import Date
+from pendulum import Date, Pendulum
 
 from .. import AbstractTestCase
 
@@ -157,8 +157,12 @@ def test_diff_for_humans_now_and_nearly_future_month(self):
         self.assertEqual('3 weeks from now', Date.today().add(weeks=3).diff_for_humans())
 
     def test_diff_for_humans_now_and_future_month(self):
-        self.assertEqual('4 weeks from now', Date.today().add(weeks=4).diff_for_humans())
-        self.assertEqual('1 month from now', Date.today().add(months=1).diff_for_humans())
+        with self.wrap_with_test_now(Pendulum.create(2016, 3, 1)):
+            self.assertEqual('4 weeks from now', Date.today().add(weeks=4).diff_for_humans())
+            self.assertEqual('1 month from now', Date.today().add(months=1).diff_for_humans())
+
+        with self.wrap_with_test_now(Pendulum.create(2017, 1, 31)):
+            self.assertEqual('1 month from now', Date.today().add(weeks=4).diff_for_humans())
 
     def test_diff_for_humans_now_and_future_months(self):
         self.assertEqual('2 months from now', Date.today().add(months=2).diff_for_humans())
@@ -191,8 +195,12 @@ def test_diff_for_humans_other_and_nearly_month(self):
         self.assertEqual('3 weeks before', Date.today().diff_for_humans(Date.today().add(weeks=3)))
 
     def test_diff_for_humans_other_and_month(self):
-        self.assertEqual('4 weeks before', Date.today().diff_for_humans(Date.today().add(weeks=4)))
-        self.assertEqual('1 month before', Date.today().diff_for_humans(Date.today().add(months=1)))
+        with self.wrap_with_test_now(Pendulum.create(2016, 3, 1)):
+            self.assertEqual('4 weeks before', Date.today().diff_for_humans(Date.today().add(weeks=4)))
+            self.assertEqual('1 month before', Date.today().diff_for_humans(Date.today().add(months=1)))
+
+        with self.wrap_with_test_now(Pendulum.create(2017, 1, 31)):
+            self.assertEqual('1 month before', Date.today().diff_for_humans(Date.today().add(weeks=4)))
 
     def test_diff_for_humans_other_and_months(self):
         self.assertEqual('2 months before', Date.today().diff_for_humans(Date.today().add(months=2)))