side_effectを使ったPythonパッチオブジェクト

Question

私は与えられた入力に基づいて特定の値を返すようにしようとしています。私はSOのいくつかの例を調べました。何らかの理由で私はまだそれを動作させることができません。ここに私が今持っているものがあります。

class EmailChecker(): 
    def is_email_correct(email): 
     some regex to determine if email is valid, returns either True or False 

def my_side_effect(**args): 
    if args[0] == '1': 
     return True 
    else: 
     return False 

myMock = mock.patch.object(EmailChecker, 'is_email_correct', side_effect=my_side_effect) 

ただし、動作していないようです。私はここで何が欠けていますか？

Answer 1

次のコードのように、デコレータまたはコンテキストマネージャとして使用patch.object：

>>> class EmailChecker(): 
...  def is_email_correct(self, email): 
...   pass 
... 
>>> def my_side_effect(*args): 
...  if args[0] == '1': 
...   return True 
...  else: 
...   return False 
... 
>>> with mock.patch.object(EmailChecker, 'is_email_correct', side_effect=my_side_effect): 
...  checker = EmailChecker() 
...  print(checker.is_email_correct('1')) 
...  print(checker.is_email_correct('2')) 
... 
True 
False 

注は：*argsと**args置き換え。 is_email_correctメソッドに引数がないので、selfが追加されました。次のように

---

my_side_effectを簡略化することができます

def my_side_effect(email): 
    return email == '1'