在Python中模拟目录结构

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了在Python中模拟目录结构，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含1209字，纯文字阅读大概需要2分钟。

内容图文

我有一些代码,我用来接受文件输入,打开和处理,然后输出一些数据.我已经使功能正常工作,我现在正在进行单元测试,下面是代码示例.

def foo(dir):
    path_to_search = join(dir, "/baz/foo")

   if isdir(path_to_search): 
   #path exists so do stuff...

        for fname in listdir(path_to_search):
            do_stuff()

   else:
       print "path doesn't exist"

我已经能够创建一个过去不容易存在的测试,但是如上所述,我断言目录结构的“/ baz / foo”部分存在(在生产中,目录结构必须具有此功能)文件,在某些情况下它不会,我们将不需要处理它.)

我试图使用TempDir和join创建一个临时目录结构,但代码总是踢出来说路径不存在.

是否可以模拟os.listdir的输出,这样我就不需要创建一个遵循所需/ baz / foo约定的临时目录结构？

解决方法:

您不需要创建虚假的目录结构,您只需要模拟isdir()和listdir()函数.

使用unittest.mock library(或外部mock library,这与Python版本< 3.3完全相同)：

try:
    # Python >= 3.3 
    from unittest import mock
except ImportError:
    # Python < 3.3
    import mock

with mock.patch('yourmodule.isdir') as mocked_isdir,         mock.patch('yourmodule.listdir') as mocked_listdir:
    mocked_isdir.return_value = True
    mocked_listdir.return_value = ['filename1', 'filename2']

    yourmodule.foo('/spam/eggs')

    mocked_isdir.assert_called_with('/spam/eggs/baz/foo')
    mocked_listdir.assert_called_with('/spam/eggs/baz/foo')

内容总结

以上是互联网集市为您收集整理的在Python中模拟目录结构全部内容，希望文章能够帮你解决在Python中模拟目录结构所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端