要在Python中使用unittest模块编写工作日的测试用例,可以按照以下步骤进行:
import unittest
from datetime import datetime
from your_module import is_weekday
class TestIsWeekday(unittest.TestCase):
def test_is_weekday(self):
# 测试日期为工作日的情况
self.assertTrue(is_weekday(datetime(2022, 1, 3))) # 星期一
self.assertTrue(is_weekday(datetime(2022, 1, 4))) # 星期二
self.assertTrue(is_weekday(datetime(2022, 1, 5))) # 星期三
self.assertTrue(is_weekday(datetime(2022, 1, 6))) # 星期四
self.assertTrue(is_weekday(datetime(2022, 1, 7))) # 星期五
# 测试日期为非工作日的情况
self.assertFalse(is_weekday(datetime(2022, 1, 1))) # 星期六
self.assertFalse(is_weekday(datetime(2022, 1, 2))) # 星期日
在测试方法中,使用self.assertTrue()
断言来验证日期是否为工作日,使用self.assertFalse()
断言来验证日期是否为非工作日。
if __name__ == '__main__':
的代码块,并在其中调用unittest.main()
运行测试。if __name__ == '__main__':
unittest.main()
完整的示例代码如下:
import unittest
from datetime import datetime
from your_module import is_weekday
class TestIsWeekday(unittest.TestCase):
def test_is_weekday(self):
# 测试日期为工作日的情况
self.assertTrue(is_weekday(datetime(2022, 1, 3))) # 星期一
self.assertTrue(is_weekday(datetime(2022, 1, 4))) # 星期二
self.assertTrue(is_weekday(datetime(2022, 1, 5))) # 星期三
self.assertTrue(is_weekday(datetime(2022, 1, 6))) # 星期四
self.assertTrue(is_weekday(datetime(2022, 1, 7))) # 星期五
# 测试日期为非工作日的情况
self.assertFalse(is_weekday(datetime(2022, 1, 1))) # 星期六
self.assertFalse(is_weekday(datetime(2022, 1, 2))) # 星期日
if __name__ == '__main__':
unittest.main()
请注意,上述代码中的your_module
应替换为你要测试的实际模块的名称,以及is_weekday
为你要测试的实际函数或类的名称。