Add main method for testing

demo.py

@@ -9,4 +9,23 @@ def add(a, b):
     Returns:
         两个数字的和
     """
-    return a + b
+    return a + b
+
+if __name__ == "__main__":
+    """测试add函数的功能"""
+    # 测试整数相加
+    print("测试整数相加:")
+    print(f"1 + 2 = {add(1, 2)}")
+    print(f"100 + 200 = {add(100, 200)}")
+    
+    # 测试浮点数相加
+    print("\n测试浮点数相加:")
+    print(f"1.5 + 2.5 = {add(1.5, 2.5)}")
+    print(f"0.1 + 0.2 = {add(0.1, 0.2)}")
+    
+    # 测试负数相加
+    print("\n测试负数相加:")
+    print(f"-1 + (-2) = {add(-1, -2)}")
+    print(f"5 + (-3) = {add(5, -3)}")
+    
+    print("\n所有测试完成！")