def test_argument_types(transform): start_date = datetime(2011, 1, 1) end_date = datetime(2020, 12, 31) holidays = USThanksgivingDay.dates(start_date, end_date) holidays2 = USThanksgivingDay.dates(transform(start_date), transform(end_date)) tm.assert_index_equal(holidays, holidays2)
def test_argument_types(transform): start_date = datetime(2011, 1, 1) end_date = datetime(2020, 12, 31) holidays = USThanksgivingDay.dates(start_date, end_date) holidays2 = USThanksgivingDay.dates( transform(start_date), transform(end_date)) tm.assert_index_equal(holidays, holidays2)
def test_argument_types(self): holidays = USThanksgivingDay.dates(self.start_date, self.end_date) holidays_1 = USThanksgivingDay.dates(self.start_date.strftime("%Y-%m-%d"), self.end_date.strftime("%Y-%m-%d")) holidays_2 = USThanksgivingDay.dates(Timestamp(self.start_date), Timestamp(self.end_date)) self.assert_index_equal(holidays, holidays_1) self.assert_index_equal(holidays, holidays_2)
def test_argument_types(self): holidays = USThanksgivingDay.dates(self.start_date, self.end_date) holidays_1 = USThanksgivingDay.dates( self.start_date.strftime('%Y-%m-%d'), self.end_date.strftime('%Y-%m-%d')) holidays_2 = USThanksgivingDay.dates(Timestamp(self.start_date), Timestamp(self.end_date)) tm.assert_index_equal(holidays, holidays_1) tm.assert_index_equal(holidays, holidays_2)
def test_usthanksgivingday(self): holidays = USThanksgivingDay.dates(self.start_date, self.end_date) holidayList = [ datetime(2011, 11, 24), datetime(2012, 11, 22), datetime(2013, 11, 28), datetime(2014, 11, 27), datetime(2015, 11, 26), datetime(2016, 11, 24), datetime(2017, 11, 23), datetime(2018, 11, 22), datetime(2019, 11, 28), datetime(2020, 11, 26), ] self.assertEqual(list(holidays), holidayList)