我是 swig 和 python 单元测试的新手。
这就是我想要做的。我有一个需要用户输入的 C++ 函数。使用 SWIG 将 C++ 代码包装到 Python 代码中。我正在尝试使用 pythons unittest 模块来模拟输入。我试过模拟 builtins.input 并用 c++ 编写我自己的基本函数,它只返回一个字符串并模拟它。
当我到达 c++ 代码中的 std::cin 时,模拟 builtins.input 仍然挂起。并模拟返回字符串的函数,不返回模拟的 return_value。
我的猜测是出于某种原因我无法模拟函数的返回值,因为它是真正的 c++ 代码而不是真正的 python。
这是我正在使用的一些示例代码:
c++ 我可以包含需要的头文件,但它真的很简单。
#include "MockTest.h"
#include <iostream>
#include <string>
MockTest::MockTest() {}
std::string MockTest::getInput()
{
std::string name;
std::cout << "Enter you name" << std::endl;
name = this->mockInput("hi"); //std::cin >> name;
std::string toReturn = "Hello " + name + " person";
return toReturn;
}
std::string MockTest::mockInput(std::string input)
{
return input;
}
swig接口(interface)文件:
%module MockTest
%include "std_string.i"
%include "std_vector.i"
%include "std_iostream.i"
%{
#include "MockTest.h"
%}
%include <MockTest.h>
python测试脚本
from unittest.mock import patch
from unittest import TestCase
import unittest
import MockTest
class Test(TestCase):
@patch('builtins.input', return_value="Will")
def test_Mocktest(self, input):
self.assertEqual(MockTest.MockTest().getInput(), 'Hello Will person')
@patch('MockTest.MockTest.mockInput', return_value="Will")
def test_Mocktest2(self, mockInput):
self.assertEqual(MockTest.MockTest().getInput(), 'Hello Will person')
if __name__ == '__main__':
unittest.main()
最佳答案
在花费了一些时间之后~一个小时在一个愚蠢的错误上:
- 生成的模块不完整这一事实(链接器 在未向其提供MockTest 对象时并未提示 undefined symbol )
- 最后生成的是因为我忘记在 MockTest.h 的类成员之前添加
public:
... X:(((
又花了一些时间调查真正的问题,我终于得出结论:问题不是因为C++语言(至少不是直接的),而是出在中介层由 Swig 创建。
详情
MockTest
类(其mockInput
方法返回值我们试图修补
)在模块MockTest
中定义>(!!! 即 MockTest.py !!!)由 Swig 自动生成(我的命令是 swig -o MockTest_wrap.cpp -python -c++ MockTest.i
).这是它的定义:
class MockTest(_object):
__swig_setmethods__ = {}
__setattr__ = lambda self, name, value: _swig_setattr(self, MockTest, name, value)
__swig_getmethods__ = {}
__getattr__ = lambda self, name: _swig_getattr(self, MockTest, name)
__repr__ = _swig_repr
def __init__(self):
this = _MockTest.new_MockTest()
try:
self.this.append(this)
except __builtin__.Exception:
self.this = this
__swig_destroy__ = _MockTest.delete_MockTest
__del__ = lambda self: None
def getInput(self):
return _MockTest.MockTest_getInput(self)
def mockInput(self, input):
return _MockTest.MockTest_mockInput(self, input)
正如您可能猜到的那样,这个 mockInput
正在被补丁
而不是来自 C++ 的那个(它的名称为 MockTest_mockInput
,它由 native 模块导出:_MockTest
- native 名称为 _wrap_MockTest_mockInput
,定义在 MocTest_wrap.cpp).当然,由于 getInput
调用 MockTest_getInput
,patch
ing mockInput
(Python 包装器) 没有任何效果(就像你修改它的主体让我们说:return "123"
)。
我准备了一些示例代码来更好地说明行为(正如我在评论中提到的,我使用的是 Python 3.5.4)。请忽略最后 3 行代码和输出,直到您阅读下一段:
from unittest.mock import patch
import MockTest
if __name__ == "__main__":
return_value = "value overridden by `patch`"
mock_input_arg = "argument passed to `MockTest.mockInput`"
mock_test = MockTest.MockTest()
with patch("MockTest.MockTest.mockInput", return_value=return_value):
print("`mock_test.getInput()` returned: \"{}\"".format(mock_test.getInput()))
print("`mock_test.mockInput(\"{}\")` returned: \"{}\"".format(mock_input_arg, mock_test.mockInput(mock_input_arg)))
print("\nDon't mind the following output for now...\n") # SAME THING about the following code
with patch("MockTest._MockTest.MockTest_mockInput", return_value=return_value):
print("`mock_test.getInput()` returned: \"{}\"".format(mock_test.getInput()))
print("`mock_test.mockInput(\"{}\")` returned: \"{}\"".format(mock_input_arg, mock_test.mockInput(mock_input_arg)))
输出:
c:\Work\Dev\StackOverflow\q45934545>"c:\Install\x64\Python\Python\3.5\python.exe" dummy.py Enter you name `mock_test.getInput()` returned: "Hello hi person" `mock_test.mockInput("argument passed to `MockTest.mockInput`")` returned: "value overridden by `patch`" Don't mind the following output for now... Enter you name: `mock_test.getInput()` returned: "Hello hi person" `mock_test.mockInput("argument passed to `MockTest.mockInput`")` returned: "value overridden by `patch`"
然后我进一步尝试补丁
MockTest._MockTest.MockTest_mockInput
,但我得到了相同的输出,因为我没有补丁
MockTest::mockInput
(来自MockTest.cpp)但_wrap_MockTest_mockInput
(来自MocTest_wrap.cpp)。
下表是 C++ 代码和 Python 之间的所有层的表,用于 mockInput
(对于 getInput
是完全一样):
当 getInput
调用 mockInput
(layer 0)时,patch
应该发生,但不幸的是,这不适用于 Python。
我想到的第一个st 解决方案是直接patch
ing getInput
(第 1 层)(如果它在许多地方使用) ).
关于python - 使用 python 包装的 c++ SWIG 模拟输入,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/45934545/