def mock_factory(): obj = ConditionFactory() obj._map = { 'type_1': mock.MagicMock(return_value=mock.sentinel.type_1), 'type_2': mock.MagicMock(return_value=mock.sentinel.type_2), 'type_3': mock.MagicMock(return_value=mock.sentinel.type_3), } return obj