Python生成指定大小的文件

转载请注明出处❤️

作者:测试蔡坨坨

原文链接:caituotuo.top/400bd75c.html


你好,我是测试蔡坨坨。

在日常测试工作中,我们经常需要对上传的文件大小进行测试,例如:一个文件上传功能,限制文件大小最大为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)

方法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)

热门相关:流鱼无恙   法医王妃不好当!   横行霸道   法医王妃不好当!   夫人你马甲又掉了