帮我生成一个API接口测试代码

API接口测试通常是通过自动化工具如Postman、JMeter、Apigee等来进行，这里我将以Python的requests库为例，展示如何编写一个简单的GET请求接口测试代码：

import requests

# API的URL
api_url = "http://example.com/api/data"

# 定义测试数据或者头部信息，如果有的话
headers = {
    'Authorization': 'Bearer your_token',  # 如果有身份验证
    'Content-Type': 'application/json'   # 根据实际接口需求设置内容类型
}

def test_api():
    try:
        # 发送GET请求
        response = requests.get(api_url, headers=headers)

        # 检查响应状态码
        if response.status_code == 200:
            print("API请求成功")
            data = response.json()  # 如果返回的是JSON，可以解析数据
            return data
        else:
            print(f"请求失败，状态码：{response.status_code}")
    except requests.exceptions.RequestException as e:
        print(f"请求出错：{e}")

test_api()