def test_days(self): test_vals = [ (1, 3), (2, 3), (3, 3), (4, 7), (5, 7), (6, 7), (7, 7), ] for i in range(8, 15): test_vals.append((i, 14),) for i in range(15, 31): test_vals.append((i, 30),) for i in range(31, 50): test_vals.append((i, -1),) test_vals.extend([(None, -1), (-1, -1)]) for days, expected_filter in test_vals: output = scraping.get_date_filter(days) self.assertEqual(output, expected_filter, msg='days_since: {}, filter: {}, expected_filter: {}'.format(days, output, expected_filter))
def test_no_last_session_returns_minus_1(self): output = scraping.get_date_filter(None) self.assertEqual(output, -1)