Python单元测试正在模拟错误的输入

所以我在练习单元测试，我有下面的代码。我第一次和第二次使用

side_effects

模拟输入，但断言失败，因为代码正在执行else语句，而不是预期的输出。这是我的运行代码

def main():
    answer = input("Would you like to create a pirate?! (yes/no)").lower()

    if answer == 'y':

        name = input("Enter the name of your pirate")
        CreateAPirate(name, "Serpent", "100")

    elif answer == 'n':
        print("Okay! you chose not to create a pirate!")
    else:
        print("You must say Yes or No")


def CreateAPirate(name, ship, age, beard="NO"):
    print("Capt:", name)
    print("Pirate Ship:", ship)
    print("Age:", age)
    print("Do you have a beard?:", beard)
    return

这是我的测试代码

import unittest
from unittest.mock import patch
import sys
from RunFile import main
import io

class TestSuite(unittest.TestCase):

    @patch('builtins.input',side_effects=['y','test'])
    def test_stdout_equals_Hello(self,m):

      saved_stdout = sys.stdout
      try:
          out = io.StringIO()
          sys.stdout = out
          main()
          output = out.getvalue().strip()
          self.assertEquals("Capt: test\nPirate Ship: Serpent\nAge: 100\nDo you have a beard?: NO",output)
      finally:
          sys.stdout = saved_stdout

if __name__ == "__main__":
    unittest.main()

最后是我的

AssertionError

代码和回溯

Failure <Click to see difference>

Traceback (most recent call last):
  File "C:\Python33\lib\unittest\mock.py", line 1087, in patched
    return func(*args, **keywargs)
  File "C:\Users\jsalce\Documents\Testsuites\IfStatements\Testsuite.py", line 18, in     test_stdout_equals_Hello
    self.assertEquals("Capt: test\nPirate Ship: Serpent\nAge: 100\nDo you have a beard?: NO",output)
AssertionError: 'Capt: test\nPirate Ship: Serpent\nAge: 100\nDo you have a beard?: NO' != 'You must say Yes or No'
+ You must say Yes or No- Capt: test
- Pirate Ship: Serpent
- Age: 100
- Do you have a beard?: NO

故障
回溯（最近一次呼叫最后一次）：
文件“C:\Python33\lib\unittest\mock.py”，第1087行，带补丁
返回函数（*参数，**键盘）
文件“C:\Users\jsalce\Documents\Testsuites\IfStatements\Testsuite.py”，第18行，在test\u stdout\u equals\u Hello中
self.assertEquals（“船长：测试\n海盗船：蛇\nAge:100\n你有胡子吗？：没有”，输出）
断言错误：“船长：测试\n海盗船：蛇\n船长：100\n你有胡子吗？：没有”！=”你必须说是或不是
+你必须说是或否-机长：测试
-海盗船：蛇
-年龄:100
-你有胡子吗？：没有

---

我知道怎么了。。我只是不知道如何修复它。

您给第一个输入字符串“yes”带来了副作用，但条件测试专门寻找字符串“y”或“n”

您可以对内置的进行更改。输入，如果您对您的船长也被命名为y感到满意，那么它将起作用。我不认为副作用是传递信息的方式

class TestSuite(unittest.TestCase):

    @patch('builtins.input')
    def test_stdout_equals_Hello(self,m):

      saved_stdout = sys.stdout
      try:
          out = io.StringIO()
          sys.stdout = out
          ori = __builtins__.input
          __builtins__.input = lambda x:'y'
          main()
          __builtins__.input = ori
          output = out.getvalue().strip()
          self.assertEquals("Capt: y\nPirate Ship: Serpent\nAge: 100\nDo you have a beard?: NO",output)
      finally:
          sys.stdout = saved_stdout

---

可能的重复并不能解决我的问题@Vor代码默认为

else

语句。不知道为什么如果我将第一个输入模拟为

'y'

副作用

应该可以正常工作。参考这篇文章

副作用

补丁

'y'

第一次调用输入，第二次调用输入“您需要调整测试用例，使其只声明您关心的文本在总输出中，而不是测试是否相等：self.assertIn（“1.Scallywag\n2.Crew\n3.Pirate\nTEST”，output）…”。stdout捕获的不仅仅是main（）的结果打印.我以前没有见过用副作用来做这件事，将来肯定会用它。我照他们说的做了，但这不是我的问题。为什么它注册了错误的输入？它默认为else语句，而它不应该。最后它又回到了那个祸害，打字错误。副作用应该是副作用。