模拟原始输入python

我有这个函数，它应该接收一个列表[x，y]

def input_origem():
    origem = raw_input("Entre com o valor de x: ")
    origem = eval(origem)
    return origem
def input_saida():
    destino = raw_input("Entre com o valor de y  ")
    destino = eval(destino)
    return destino


def my func(origem, destino):

..
...
code 
..
...

print  myfunc(input_origem(), input_saida())

这段代码运行得很好，但我不知道如何模拟这个函数

我试着这样做：

class TEste(base.TestCase):
    @base.TestCase.mock.patch('mypath.input_origem')
    @base.TestCase.mock.patch('mypath.input_saida')
    def test_movimento_cavalo(self, origem_mock, saida_mock):
        origem_mock = self.mock.MagicMock()
        saida_mock = self.mock.MagickMock()
        myfunc(origem_mock, saida_mock)
        myfunc.should.be.equal([1,1])

但测试从未完成。当我取消时，我得到：

origem=原始输入（“Entre com o valor de x:”） 键盘中断

---

您应该能够使用您通常用来模拟事物的任何工具来模拟

\uuuuuu内置\uuuuuuu.raw\u输入

使用
unittest
和
mock
的一个非常简单的示例如下所示：

import unittest
import mock
import __builtin__

def test_raw_input():
    return raw_input()

class Test(unittest.TestCase):
    @mock.patch.object(__builtin__, 'raw_input')
    def test_stuff(self, mock_raw_input):
        mock_raw_input.return_value = 7
        self.assertEqual(test_raw_input(), 7)

unittest.main()

你能提供一些例子吗？@PythonMaster——添加了一个简单的例子。