好的，首先，我觉得有必要指出，在原始代码中，实际上需要解决两个问题：raw_input (输入副作用)需要被嘲笑。print (输出副作用)需要检查。在理想的单元测试功能中，不会有副作用。简单地通过传递参数来测试函数，然后检查其输出。但是通常我们想在像您这样的函数中测试不理想的函数IE。那么我们该怎么办呢？好吧，在Python 3.3中，我上面列出的两个问题变得微不足道，因为该unittest模块获得了模拟和检查副作用的功能。但是，从2014年初开始，只有30％的Python程序员开始使用3.x，因此，为了使另外70％的Python程序员仍在使用2.x，我将概述一个答案。以目前的速度，直到2019年，3.x才会超过2.x，直到2027年，2.x才会消失。因此，我认为这个答案将在未来几年中有用。我想一次解决上面列出的问题，因此我将首先将您的功能从使用print作为输出更改为使用return。毫不奇怪，这是代码：def answerReturn(): ans = raw_input('enter yes or no') if ans == 'yes': return 'you entered yes' if ans == 'no': return 'you entered no'因此，我们要做的只是模拟raw_input。足够容易-Omid Raha对这个问题的回答向我们展示了如何通过__builtins__.raw_input模拟实现来淘汰实现。除了他的答案没有正确地组织为TestCase和功能外，我将对此进行演示。
import unittest class TestAnswerReturn(unittest.TestCase): def testYes(self): original_raw_input = builtins.raw_input builtins.raw_input = lambda _: 'yes' self.assertEqual(answerReturn(), 'you entered yes') builtins.raw_input = original_raw_input def testNo(self): original_raw_input = builtins.raw_input builtins.raw_input = lambda _: 'no' self.assertEqual(answerReturn(), 'you entered no') builtins.raw_input = original_raw_input关于Python命名约定的小注意事项-解析器需要但未使用的变量通常被命名_，例如在lambda的未使用变量的情况下(通常情况下，提示是显示给用户，如果raw_input您使用'想知道为什么在这种情况下根本不需要它)。无论如何，这是混乱且多余的。因此，我将通过添加来消除重复contextmanager，这将允许使用简单的with语句。
from contextlib import contextmanager@contextmanagerdef mockRawInput(mock): original_raw_input = builtins.raw_input builtins.raw_input = lambda _: mock yield builtins.raw_input = original_raw_inputclass TestAnswerReturn(unittest.TestCase): def testYes(self): with mockRawInput('yes'): self.assertEqual(answerReturn(), 'you entered yes') def testNo(self): with mockRawInput('no'): self.assertEqual(answerReturn(), 'you entered no')
我认为这很好地回答了本文的第一部分。第二部分进行检查print。我发现这比较棘手-我很想听听是否有人有更好的答案。无论如何，该print语句不能被覆盖，但是如果您改用print()函数(如应有的话)，则from future import print_function可以使用以下语句：
class PromiseString(str): def set(self, newString): self.innerString = newString def eq(self, other): return self.innerString == other@contextmanagerdef getPrint(): promise = PromiseString() original_print = builtin.print builtin.print = lambda message: promise.set(message) yield promise builtin.print = original_printclass TestAnswer(unittest.TestCase): def testYes(self): with mockRawInput('yes'), getPrint() as response: answer() self.assertEqual(response, 'you entered yes') def testNo(self): with mockRawInput('no'), getPrint() as response: answer() self.assertEqual(response, 'you entered no')
棘手的是，您需要yield在输入with块之前做出响应。但是，直到调用块print()内部，您才能知道该响应是什么with。如果字符串是可变的，那会很好，但不是。因此，做了一个小的Promise或proxy类PromiseString。它仅做两件事-允许设置一个字符串(或其他任何东西)，并让我们知道它是否等于另一个字符串。阿PromiseString是yieldED，然后设置为通常将值print的内with块。希望您能欣赏我今晚花了大约90分钟时间整理而成的所有技巧。我测试了所有这些代码，并验证了它们在Python 2.7中都可以使用。
