ebc_2in2crc avatar ebc_2in2crc committed c0b0f68

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

Comments (0)

Files changed (2)

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>で置換されるはず")
 
         return False
     
     def censor(self, msg):
-        return msg.replace(self.wordlist, "<censored>")
+        for word in self.wordlist:
+            msg = msg.replace(word, "<censored>")
+        return msg
Tip: Filter by directory path e.g. /media app.js to search for public/media/app.js.
Tip: Use camelCasing e.g. ProjME to search for ProjectModifiedEvent.java.
Tip: Filter by extension type e.g. /repo .js to search for all .js files in the /repo directory.
Tip: Separate your search with spaces e.g. /ssh pom.xml to search for src/ssh/pom.xml.
Tip: Use ↑ and ↓ arrow keys to navigate and return to view the file.
Tip: You can also navigate files with Ctrl+j (next) and Ctrl+k (previous) and view the file with Ctrl+o.
Tip: You can also navigate files with Alt+j (next) and Alt+k (previous) and view the file with Alt+o.