def test_formatweek(self): """ Test the formatweek function. """ today = TODAY cal = FedocalCalendar(today.year, today.month, today.day) output = cal.formatweek([(1, 1), (2, 2)]).replace(' today', '') self.assertEqual( output, '<tr><td class="tue">1</td><td class="wed">2</td></tr>') self.assertEqual( cal.formatweek([(0, 1), (1, 2)]).replace(' today', ''), '<tr><td class="noday"> </td><td class="wed">1</td></tr>') output = cal.formatweek([(1, 1), (2, 2)], True).replace(' today', '') self.assertEqual( output, '<tr class="current_week"><td class="tue">1</td><td ' 'class="wed">2</td></tr>')
def test_formatweek(self): """ Test the formatweek function. """ today = TODAY cal = FedocalCalendar(today.year, today.month, today.day) output = cal.formatweek([(1, 1), (2, 2)]).replace(' today', '') self.assertEqual( output, '<tr><td class="tue">1</td><td class="wed">2</td></tr>') self.assertEqual( cal.formatweek([(0, 1), (1, 2)]), '<tr><td class="noday"> </td><td class="wed">1</td></tr>') output = cal.formatweek([(1, 1), (2, 2)], True).replace(' today', '') self.assertEqual( output, '<tr class="current_week"><td class="tue">1</td><td '\ 'class="wed">2</td></tr>')