def test_invalid_quarters(self, year, quarter): with self.assertRaises(AssertionError): get_quarter_date_range(year, quarter)
def test_quarter_ranges(self, year, quarter, expected_start, expected_end): startdate, enddate = get_quarter_date_range(year, quarter) self.assertEqual(expected_start, startdate) self.assertEqual(expected_end, enddate)
def get_quarter(year, quarter): return DateSpan(*get_quarter_date_range(year, quarter), inclusive=False)