def test_paid(self): user = User() self.assertFalse(user.is_paid) self.assertEqual(user.paid_days_left(), 0) user.add_paid_time(datetime.timedelta(days=1)) self.assertTrue(user.is_paid) self.assertEqual(user.paid_days_left(), 1) user.paid_until = datetime.datetime.fromtimestamp(1) user.add_paid_time(datetime.timedelta(days=1)) self.assertTrue(user.is_paid) self.assertEqual(user.paid_days_left(), 1)