单元测试在控制台中询问两个用户输入()的方法

  • 时间:
  • 分类: Python

我有一个小的控制台应用程序，通过要求用户输入两个数字来添加两个数字：

选择 | 换行 | 行号
  1.      def add():
  2.          a = int(input('Enter first number '))
  3.          b= int(input('Enter second number '))
  4.          return a + b

如何进行单元测试？我尝试了以下操作，但它总是返回false，即使它不要求用户输入，这很好：2.3.4.5.6.7.8.9

选择 | 换行 | 行号
  1. import unittest
  2.  
  3. from unittest.mock import patch
  4.  
  5.  
  6. @patch('builtins.input', return_value='2')
  7. @patch('builtins.input', return_value='3')
  8. def test_add(self,  a, b ):
  9.     self.assertEqual(result, 5)

标签: python

添加新评论