第一个文件中，可以创建一个函数，接受参数，然后使用参数从网络上下载数据。第二个文件中，可以创建一个测试函数，调用第一个函数，传入参数，并检查函数的输出结果。 点击复制后，将打开ChatGPT体验页。帮我写出这两个文件的python代码

第一个文件中：def download_data(param):
# code to download data from the internet based on the parameter第二个文件中：def test_function():
# call the download_data function
data = download_data(param)
# check the output result
assert data == expected_result