V2EX = way to explore
V2EX 是一个关于分享和探索的地方
现在注册
已注册用户请  登录
推荐学习书目
Learn Python the Hard Way
Python Sites
PyPI - Python Package Index
http://diveintopython.org/toc/index.html
Pocoo
值得关注的项目
PyPy
Celery
Jinja2
Read the Docs
gevent
pyenv
virtualenv
Stackless Python
Beautiful Soup
结巴中文分词
Green Unicorn
Sentry
Shovel
Pyflakes
pytest
Python 编程
pep8 Checker
Styles
PEP 8
Google Python Style Guide
Code Style from The Hitchhiker's Guide
patx
V2EX  ›  Python

Python 能否 mock 函数,只获取入参或返回值,不改变函数逻辑?

  •  
  •   patx · 2022-02-12 21:08:20 +08:00 · 2945 次点击
    这是一个创建于 1016 天前的主题,其中的信息可能已经有所发展或是发生改变。

    对 Python 不是很熟悉,碰到这种情况,比如:

    函数 func1 有文件路径的入参,func1 是写入内容到该文件。

    我想写一个单元测试,只想获取入参或者返回值,做后续断言用,但是不覆盖 func1 的功能。

    我看了一下 Mock side_effect 之类的属性,是把函数功能覆盖掉了,有没有更好的方法?

    7 条回复    2022-02-12 22:23:37 +08:00
    lanyi96
        1
    lanyi96  
       2022-02-12 21:17:54 +08:00
    装饰器?
    patx
        2
    patx  
    OP
       2022-02-12 21:22:34 +08:00
    @lanyi96 不改动该函数的前提下做单元测试,可以 mock 装饰器?
    imn1
        3
    imn1  
       2022-02-12 21:33:07 +08:00
    装饰器可以获取并继续跑原函数,在装饰器内输出到其他地方(全局变量或外存储)就行了

    如果可以修改原来代码,加个装饰器就可以了
    不能的话就继承过来再加

    可以看看 contextlib 和 inspect ,或许有用
    saber000
        4
    saber000  
       2022-02-12 21:57:15 +08:00   ❤️ 1
    side_effect 可以再指向原函数呀
    Trim21
        5
    Trim21  
       2022-02-12 22:05:12 +08:00   ❤️ 1
    mock.Mock(wraps=original_object)
    patx
        6
    patx  
    OP
       2022-02-12 22:21:37 +08:00
    @Trim21 感谢,问题解决了
    patx
        7
    patx  
    OP
       2022-02-12 22:23:37 +08:00
    @imn1 有局限性,因为 func1 可能是第三方包的函数。我用 patch 的 wraps 属性解决了
    关于   ·   帮助文档   ·   博客   ·   API   ·   FAQ   ·   实用小工具   ·   3827 人在线   最高记录 6679   ·     Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 · 24ms · UTC 10:34 · PVG 18:34 · LAX 02:34 · JFK 05:34
    Developed with CodeLauncher
    ♥ Do have faith in what you're doing.