如何更改python函数变量的值以进行单元测试

def my_function():
    x = 5
    return x

# 单元测试
def test_my_function():
    assert my_function() == 5

    # 更改变量的值
    my_function.x = 10

    assert my_function() == 10

test_my_function()

from unittest import mock

def my_function():
    x = 5
    return x

# 单元测试
def test_my_function():
    assert my_function() == 5

    # 使用mock库替换变量的值
    with mock.patch('__main__.my_function') as mock_function:
        mock_function.x = 10
        assert my_function() == 10

test_my_function()

class MyClass:
    x = 5

def my_function():
    return MyClass.x

# 单元测试
def test_my_function():
    assert my_function() == 5

    # 修改类的属性
    MyClass.x = 10

    assert my_function() == 10

test_my_function()