Python生成指定大小的文件

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

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

方法1:

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

  1. # author: 咖啡加剁椒

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

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

  4. def generate_file(file_path, file_size_bytes):

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

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

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

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

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

  10.         for _ in range(repetitions):

  11.             file.write(text)

  12.         if remainder > 0:

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

  14. if __name__ == '__main__':

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

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

方法2:

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

  1. # author: 咖啡加剁椒

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

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

  4. import os

  5. def generate_file(file_path, file_size_bytes):

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

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

  8. if __name__ == '__main__':

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

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

以上,完。

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

最后感谢每一个认真阅读我文章的人!作为一位过来人也是希望大家少走一些弯路,如果你不想再体验一次学习时找不到资料,坚持几天便放弃的感受的话,在这里我给大家分享一些软件测试的学习资源,这些资料,对于【软件测试】的朋友来说应该是最全面最完整的备战仓库,希望能给你前进的路上带来帮助。如果你用得到的话可以直接拿走:

软件测试资料领取:[内部资源] 想拿年薪40W+的软件测试人员,这份资料必须领取~

软件测试面试刷题工具领取:软件测试面试刷题【800道面试题+答案免费刷】

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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值