Как указать возвращаемое значение по умолчанию методу, который вызывается во время модульного теста? – Python

Вот пример класса, который упрощает то, что у меня есть:

class.py

class MyClass(object): @staticmethod def getDictionary(): #some calculations, returns a dictionary def checkConfiguration(self): #some code self.getDictionary() #some other code return value 

И теперь я делаю единичный тест для checkConfiguration :

classTest.py

 import class import unittest class TestMyClass(unittest.TestCase): def setUp(self): self.classTest = class.MyClass() def test_CheckConfiguration(self): #What to put here? 

Исходный CheckConfiguration вызывает getDictionary . Есть ли способ сообщить test_CheckConfiguration(self) что если getDictionary вызывается, он должен автоматически возвращать словарь, который я могу ввести? Что-то вроде:

  def test_CheckConfiguration(self): if method getDictionary is called: return {'a':123, 'b':22} checkValue = self.classTest.checkConfiguration() 

Я знаю, что это возможно на Java, хотя у меня недостаточно опыта и в этом. Спасибо.

• Есть ли один код строки, чтобы найти максимальное значение в матрице?
• Добавление элементов из списка в словарь в Python
• Есть ли способ Pythonic для итерации по «расширенному» списку источников?
• Как передать класс в качестве параметра и наследовать методы
• Python NameError, когда var определенно определен

 import class import pymock import unittest class TestMyClass(pymock.PyMockTestCase): def setUp(self): self.classTest = class.MyClass() def test_CheckConfiguration(self): self.override(self.classTest, 'getDictionary') pymock.expectAndReturn(self.classTest.getDictionary(), {'a':123, 'b':22}) self.replay() checkValue = self.classTest.checkConfiguration() self.verify()