如何使用模拟来测试next_day_of_week函数

import unittest
from unittest.mock import Mock
from datetime import datetime, timedelta
from your_module import next_day_of_week

class TestNextDayOfWeek(unittest.TestCase):
    def test_next_day_of_week(self):
        # 创建一个模拟的datetime对象
        mock_datetime = Mock(spec=datetime)
        mock_datetime.weekday.return_value = 0  # 模拟一个星期一的日期

        # 使用模拟的datetime对象调用next_day_of_week函数
        result = next_day_of_week(mock_datetime, 1)

        # 检查结果是否正确
        expected_result = mock_datetime + timedelta(days=1)
        self.assertEqual(result, expected_result)

if __name__ == '__main__':
    unittest.main()