Python生成指定大小的文件

在日常测试工作中，我们经常需要对上传的文件大小进行测试，例如：一个文件上传功能，限制文件大小最大为10MB，此时我们可能需要测试10MB以及其边界值9MB和11MB；再或者我们有时需要测试一个超大文件，进行大文件的测试。

针对以上情况，可能一时难以找到符合准确数据的测试文件，这时就可以使用Python来帮助我们生成任意大小的文件，这里提供两种解决方案。

方法1：

使用特定大小的文本重复生成，指定一个文本字符串text，然后将其重复复制直至达到所需的文件大小。

1. # author: 咖啡加剁椒

2. # datetime: 2023/6/8 1:31

3. # function: 使用特定大小的文本生成指定大小的文件

6. def generate_file(file_path, file_size_bytes):

7.     text = "This is some sample text by caituotuo."  # 要重复的文本

8.     text_size_bytes = len(text.encode('utf-8'))  # 每个重复的文本的大小（以字节为单位）

10.     repetitions = file_size_bytes // text_size_bytes  # 需要重复的次数

11.     remainder = file_size_bytes % text_size_bytes  # 剩余的字节数

13.     with open(file_path, 'w') as file:

14.         for _ in range(repetitions):

15.             file.write(text)

17.         if remainder > 0:

18.             file.write(text[:remainder])

21. if __name__ == '__main__':

22.     # 生成一个大小为10MB的PDF文件

23.     generate_file('caituotuo.pdf', 1024 * 1024 * 10)

方法2：

使用特定大小的随机数生成，使用随机数生成器生成特定大小的字节，并将其写入文件中。

1. # author: 咖啡加剁椒

2. # datetime: 2023/6/8 2:31

3. # function: 使用特定大小的随机数生成文件

5. import os

8. def generate_file(file_path, file_size_bytes):

9.     with open(file_path, 'wb') as file:

10.         file.write(os.urandom(file_size_bytes))

13. if __name__ == '__main__':

14.     # 生成一个大小为1MB的随机数据文件

15.     generate_file('caituotuo.docx', 1024 * 1024)

以上，完。

脚踏实地，仰望星空，和小编一起学习软件测试，升职加薪！

行动吧，在路上总比一直观望的要好，未来的你肯定会感 谢现在拼搏的自己！如果想学习提升找不到资料，没人答疑解惑时，请及时加入扣群： 320231853，里面有各种软件测试+开发资料和技术可以一起交流学习哦。

最后感谢每一个认真阅读我文章的人，礼尚往来总是要有的，虽然不是什么很值钱的东西，如果你用得到的话可以直接拿走：

​

这些资料，对于【软件测试】的朋友来说应该是最全面最完整的备战仓库，这个仓库也陪伴上万个测试工程师们走过最艰难的路程，希望也能帮助到你！