import os
def merge_files(files):
merged_file_name = "merged.txt"
with open(merged_file_name, "w") as merged_file:
for file in files:
with open(file, "r") as f:
merged_file.write(f.read())
os.remove(file)
merge_files(["file1.txt", "file2.txt"])
def test_merge_files():
merged_file_name = "merged.txt"
merge_files(["file1.txt", "file2.txt"])
assert os.path.isfile(merged_file_name), "Merged file doesn't exist"
with open(merged_file_name, 'r') as merged_file:
content = merged_file.read()
assert content == "file1textfile2text", "Merged content doesn't match"
test_merge_files()
在这个示例中,我们编写了两个函数。第一个函数是“merge_files”,它接受一个文件名列表并将它们合并为一个文件。我们创建一个名为“merged.txt”的文件,然后逐个读取所有传递的文件并将其内容写入到merged.txt中。最后,我们使用“os.remove”删除所有传递的文件。
第二个函数是“test_merge_files”。这个函数测试是否成功合并文件。我们首先调用“merge_files”函数,然后验证创建一个名为“merged.txt”的文件。接着,我们读取这个文件并验证它包含了所有文件的内容。如果任意的验证失败,测试函数将会引发AssertionError。
通过这两个函数,我们可以测试和验证合并文件的函数已经成功地编写了,而不需要返回任何结果。