python简单测试代码

以下是一个简单的Python测试代码，可以用于测试一个函数：

def test_function():
    assert function_name(input1) == expected_output1
    assert function_name(input2) == expected_output2
    assert function_name(input3) == expected_output3
    print("All test cases pass")

test_function()

在上面的代码中，`function_name` 是你要测试的函数名，`input1`、`input2`、`input3` 是不同的输入值，`expected_output1`、`expected_output2`、`expected_output3` 是你预期得到的输出值。

如果函数的输出与预期的输出不符，assert语句将抛出AssertionError异常。如果所有测试用例都通过，将打印"All test cases pass"。