無法理解如何使用 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()