Commits

ebc_2in2crc  committed c0b0f68

censorメソッドが複数NGワードに対応していない問題に対応しました

  • Participants
  • Parent commits 91cfab9

Comments (0)

Files changed (2)

File test_wordfilter.py

         self.assertFalse(wordfilter.detect(u"t_wada:昨日のArsenal vs Chelsea 熱かった!"))
 
     def test_censor_unmatch(self):
-        wordfilter = WordFilter("Arsenal")
+        wordfilter = WordFilter(["Arsenal"])
         msg = "t_wada:昨日 vs Chelsea 熱かった!"
         self.assertEqual(msg, wordfilter.censor(msg), "そのままの文字列が返ってくるはず")
 
     def test_censor_match(self):
-        wordfilter = WordFilter("Arsenal")
+        wordfilter = WordFilter(["Arsenal"])
         msg = "t_wada:昨日Arsenal vs Chelsea 熱かった!"
         self.assertEqual("t_wada:昨日<censored> vs Chelsea 熱かった!", wordfilter.censor(msg), "<censored>で置換されるはず")
 

File wordfilter.py

         return False
     
     def censor(self, msg):
-        return msg.replace(self.wordlist, "<censored>")
+        for word in self.wordlist:
+            msg = msg.replace(word, "<censored>")
+        return msg