无法理解如何使用unittest模块在python中编写工作日的测试用例

Question

我在python中有一个函数，它以工作日为参数，在工作日成功时返回一个int，我是python单元测试的新手，我想知道如何编写单元测试用例。下面是函数：

def dayNameFromWeekday(weekday):
    if weekday == "Monday":
        return 0
    if weekday == "Tuesday":
        return 1
    if weekday == "Wednesday":
        return 2
    if weekday == "Thursday":
        return 3
    if weekday == "Friday":
        return 4
    if weekday == "Saturday":
        return 5
    if weekday == "Sunday":
        return 6

这是测试文件

class Utils(unittest.TestCase):
    def test_dayNameFromWeekday(self, weekday):
        if weekday == "Monday":
            return 0
        elif weekday == "Tuesday":
            return 1
        elif weekday == "Wednesday":
            return 2
        elif weekday == "Thursday":
            return 3
        elif weekday == "Friday":
            return 4
        elif weekday == "Saturday":
            return 5
        elif weekday == "Sunday":
            return 6
        else:
            return -1  # Any error message

    res = self.test_dayNameFromWeekday("Monday")

而且我不知道如何为这个函数编写测试用例，请告诉我在unittest python中做事情的方式。

Answer 1

以下是解决方案：

import unittest
class Utils(unittest.TestCase):
    def test_dayNameFromWeekday(self, weekday):
        if weekday == "Monday":
            return 0
        elif weekday == "Tuesday":
            return 1
        elif weekday == "Wednesday":
            return 2
        elif weekday == "Thursday":
            return 3
        elif weekday == "Friday":
            return 4
        elif weekday == "Saturday":
            return 5
        elif weekday == "Sunday":
            return 6
        else:
            return -1  # Any error message
    def runTest(self):
        res = self.test_dayNameFromWeekday("Monday")
        print(res)

res = Utils()
res.run()