Как мне переопределить функции в Python?

Question

Я получил функцию в определенном модуле, которую я хочу переопределить (макет) во время выполнения для целей тестирования. Насколько я понимаю, определение функции - это не что иное, как присваивание в python (само определение модуля является своего рода выполняемой функцией). Как я уже сказал, я хочу сделать это в настройках тестового примера, чтобы функция, которую нужно переопределить, находилась в другом модуле. Каков синтаксис для этого? Например, 'module1' - это мой модуль, а 'func1' - моя функция, в моем тестовом примере я пробовал это (безуспешно):

import module1

module1.func1 = lambda x: return True

Answer 1

import module1
import unittest

class MyTest(unittest.TestCase):
    def setUp(self):
        # Replace othermod.function with our own mock
        self.old_func1 = module1.func1
        module1.func1 = self.my_new_func1

    def tearDown(self):
        module1.func1 = self.old_func1

    def my_new_func1(self, x):
        """A mock othermod.function just for our tests."""
        return True

    def test_func1(self):
        module1.func1("arg1")

Множество библиотек насмешек предоставляют инструменты для такого рода насмешек, вы должны изучить их, так как вы, вероятно, получите от них большую помощь.

Answer 2

import foo

def bar(x):
    pass

foo.bar = bar

Answer 3

Просто назначьте новую функцию или лямбду старому имени:

>>> def f(x):
...     return x+1
... 
>>> f(3)
4
>>> def new_f(x):
...     return x-1
... 
>>> f = new_f
>>> f(3)
2

Работает также, когда функция из другого модуля:

### In other.py:
# def f(x):
#    return x+1
###

import other

other.f = lambda x: x-1

print other.f(1)   # prints 0, not 2

Answer 4

Использовать переопределение: http://github.com/joeheyming/redef

import module1
from redef import redef

rd_f1 = redef(module1, 'func1', lambda x: True)

Когда rd_f1 выходит из области видимости или удаляется, func1 возвращается к нормальному состоянию

Answer 5

Если вы хотите перезагрузить файл интерпретатора foo.py, который вы редактируете, вы можете создать функцию простого типа и использовать execfile (), но я только что узнал, что она не работает без глобального списка.из всех функций (к сожалению), если у кого-то нет лучшего представления:

Где-то в файле foo.py:

def refoo ():
  global fooFun1, fooFun2
  execfile("foo.py")

В интерпретаторе python:

refoo () # Теперь у вас есть последние изменения от foo.py