缩短经常使用的代码段以测试Python中的返回值

def test_function():
    # 您的函数代码
    return value

def test_return_value():
    assert test_function() == expected_value, f"Expected {expected_value}, but got {test_function()}"

test_return_value()

import unittest

class TestFunction(unittest.TestCase):
    def test_return_value(self):
        self.assertEqual(test_function(), expected_value, f"Expected {expected_value}, but got {test_function()}")

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