Python生成指定大小的文件

小灰灰 2023-10-12 11:56 80阅读 0赞

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

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

**方法1：**

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

# author: 爱吃香菜
    # datetime: 2023/6/8 1:31
    # function: 使用特定大小的文本生成指定大小的文件
    
    
    def generate_file(file_path, file_size_bytes):
        text = "This is some sample text by caituotuo."  # 要重复的文本
        text_size_bytes = len(text.encode('utf-8'))  # 每个重复的文本的大小（以字节为单位）
    
        repetitions = file_size_bytes // text_size_bytes  # 需要重复的次数
        remainder = file_size_bytes % text_size_bytes  # 剩余的字节数
    
        with open(file_path, 'w') as file:
            for _ in range(repetitions):
                file.write(text)
    
            if remainder > 0:
                file.write(text[:remainder])
    
    
    if __name__ == '__main__':
        # 生成一个大小为10MB的PDF文件
        generate_file('caituotuo.pdf', 1024 * 1024 * 10)

![85f9f4ed44bd4528aed9e8c5f8a1b1f0.png][]

**方法2：**

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

# author: 爱吃香菜
    # datetime: 2023/6/8 2:31
    # function: 使用特定大小的随机数生成文件
    
    import os
    
    
    def generate_file(file_path, file_size_bytes):
        with open(file_path, 'wb') as file:
            file.write(os.urandom(file_size_bytes))
    
    
    if __name__ == '__main__':
        # 生成一个大小为1MB的随机数据文件
        generate_file('caituotuo.docx', 1024 * 1024)

![b9caebaf386a4603bafafce8c216d014.png][]

以上，完。

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

**最后：**下方这份完整的软件测试视频教程已经整理上传完成，需要的朋友们可以自行领取**【保证100%免费】**

![7d4a32aca79045099aee747f5bcd0628.png][]

#### **整套资料获取** ####

![509173cee77d4b93a4b396b7b4cf5f0b.gif][]

[85f9f4ed44bd4528aed9e8c5f8a1b1f0.png]: https://img-blog.csdnimg.cn/85f9f4ed44bd4528aed9e8c5f8a1b1f0.png
[b9caebaf386a4603bafafce8c216d014.png]: https://img-blog.csdnimg.cn/b9caebaf386a4603bafafce8c216d014.png
[7d4a32aca79045099aee747f5bcd0628.png]: https://img-blog.csdnimg.cn/7d4a32aca79045099aee747f5bcd0628.png
[509173cee77d4b93a4b396b7b4cf5f0b.gif]: https://img-blog.csdnimg.cn/509173cee77d4b93a4b396b7b4cf5f0b.gif