Add divide function to test.py

test.py

@@ -12,10 +12,38 @@ def multiply(a, b):
     return a * b
 
 
+def divide(a, b):
+    """
+    计算两个数字相除并返回结果。
+
+    参数:
+    a (int/float): 被除数
+    b (int/float): 除数
+
+    返回:
+    float: a除以b的结果
+    异常:
+    ValueError: 当除数为0时抛出
+    """
+    if b == 0:
+        raise ValueError("除数不能为0")
+    return a / b
+
+
 if __name__ == "__main__":
     # 测试代码
     print("测试 multiply 函数:")
     print(f"2 * 3 = {multiply(2, 3)}")
     print(f"4.5 * 2 = {multiply(4.5, 2)}")
     print(f"-5 * 6 = {multiply(-5, 6)}")
-    print(f"0 * 100 = {multiply(0, 100)}")
+    print(f"0 * 100 = {multiply(0, 100)}")
+
+    print("\n测试 divide 函数:")
+    try:
+        print(f"6 / 3 = {divide(6, 3)}")
+        print(f"10 / 2 = {divide(10, 2)}")
+        print(f"5.5 / 2 = {divide(5.5, 2)}")
+        print(f"0 / 5 = {divide(0, 5)}")
+        print(f"6 / 0 = {divide(6, 0)}")
+    except ValueError as e:
+        print(f"错误: {e}")