debian-workalendar/workalendar/tests/__init__.py

31 lines
691 B
Python

import warnings
from datetime import date
from unittest import TestCase
from ..core import Calendar
class CoreCalendarTest(TestCase):
cal_class = Calendar
def setUp(self):
super().setUp()
self.year = date.today().year
self.cal = self.cal_class()
class GenericCalendarTest(CoreCalendarTest):
def setUp(self):
super().setUp()
warnings.simplefilter('ignore')
def test_weekend_days(self):
class_name = self.cal_class.__name__
if class_name in ('Calendar',):
return
try:
self.cal.get_weekend_days()
except NotImplementedError:
assert False, (self.cal, class_name)