def test_periods(self): cohort = self.mox.CreateMockAnything() cohort.users = User.objects.all() self.mox.ReplayAll() periods = list(Period.periods(cohort, [60, 30, 90])) for period in periods: self.assertEqual(period.cohort, cohort) self.assertEqual(periods[0].start_day, 1) self.assertEqual(periods[0].end_day, 30) self.assertEqual(periods[1].start_day, 30) self.assertEqual(periods[1].end_day, 60) self.assertEqual(periods[2].start_day, 60) self.assertEqual(periods[2].end_day, 90) self.mox.VerifyAll()