skorotkiewicz/rDrama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Compact slurs dict and added support for word censor

Browse source
This commit is contained in:
Yo Mama 2021-10-16 20:42:00 +02:00
parent 3fe170f012
commit 0de6608408
3 changed files with 91 additions and 150 deletions
Download patch file Download diff file Expand all files Collapse all files

43
test/files/helpers/test_word_censor.py
View file

@ -7,6 +7,16 @@ from files.helpers import word_censor
from files.helpers.word_censor import create_variations_slur_regex, create_replace_map, censor_slurs, sub_matcher
def test_create_variations_slur_regex_for_slur_with_spaces():
expected = [r"(\s|>)(retard)(\s|<)",
r"(\s|>)(Retard)(\s|<)",
r"(\s|>)(RETARD)(\s|<)"]
result = create_variations_slur_regex(" retard ")
assert_that(result).is_length(3).contains_only(*expected)
def test_create_variations_slur_regex_single_word():
expected = [r"(\s|>)(retard)|(retard)(\s|<)",
r"(\s|>)(Retard)|(Retard)(\s|<)",
@ -31,6 +41,7 @@ def test_create_variations_slur_regex_multiple_word():
"tranny": "🚂🚃🚃",
"kill yourself": "keep yourself safe",
"faggot": "cute twink",
" nig ": "🏀",
})
def test_create_replace_map():
expected = {
@ -44,13 +55,16 @@ def test_create_replace_map():
"faggot": "cute twink",
"Faggot": "Cute twink",
"FAGGOT": "CUTE TWINK",
"nig": "🏀",
"Nig": "🏀",
"NIG": "🏀",
}
result = create_replace_map()
assert_that(result).is_equal_to(expected)
@patch("files.helpers.word_censor.REPLACE_MAP", {'retard': 'r-slur'})
@patch("files.helpers.word_censor.REPLACE_MAP", {'retard': 'r-slur', 'NIG': '🏀'})
def test_sub_matcher():
match = re.search(r"(\s|>)(retard)|(retard)(\s|<)", "<p>retard</p>")
assert_that(sub_matcher(match)).is_equal_to(">r-slur")
@ -58,8 +72,14 @@ def test_sub_matcher():
match = re.search(r"(\s|>)(retard)|(retard)(\s|<)", "<p>noretard</p>")
assert_that(sub_matcher(match)).is_equal_to("r-slur<")
match = re.search(r"(\s|>)(NIG)(\s|<)", "<p>NIG</p>")
assert_that(sub_matcher(match)).is_equal_to(">🏀<")
@patch("files.helpers.word_censor.SLURS", {'retard': 'r-slur', 'manlet': 'little king'})
match = re.search(r"(\s|>)(NIG)(\s|<)", "<p>NIG </p>")
assert_that(sub_matcher(match)).is_equal_to(">🏀 ")
@patch("files.helpers.word_censor.SLURS", {'retard': 'r-slur', 'manlet': 'little king', ' nig ': '🏀'})
def test_censor_slurs():
word_censor.REPLACE_MAP = create_replace_map()
@ -68,13 +88,30 @@ def test_censor_slurs():
assert_that(censor_slurs(None, "that is Retarded like")).is_equal_to("that is R-slured like")
assert_that(censor_slurs(None, "that is SUPERRETARD like")).is_equal_to("that is SUPERR-SLUR like")
assert_that(censor_slurs(None, "<p>Manlets get out!</p>")).is_equal_to("<p>Little kings get out!</p>")
assert_that(censor_slurs(None, '... "retard" ...')).is_equal_to('... "retard" ...')
assert_that(censor_slurs(None, '... ReTaRd ...')).is_equal_to('... ReTaRd ...')
assert_that(censor_slurs(None, '... aretarded ...')).is_equal_to('... aretarded ...')
assert_that(censor_slurs(None, '... xretardx ...')).is_equal_to('... xretardx ...')
assert_that(censor_slurs(None, "LLM is a manlet hehe")).is_equal_to("LLM is a little king hehe")
assert_that(censor_slurs(None, "LLM is :marseycapitalistmanlet: hehe")) \
.is_equal_to("LLM is :marseycapitalistmanlet: hehe")
assert_that(censor_slurs(None, '... Nig ...')).is_equal_to('... 🏀 ...')
assert_that(censor_slurs(None, '<p>NIG</p>')).is_equal_to('<p>🏀</p>')
assert_that(censor_slurs(None, '... nigeria ...')).is_equal_to('... nigeria ...')
assert_that(censor_slurs(None, "<p>retarded SuperManlet NIG</p>")) \
.is_equal_to("<p>r-slured SuperLittle king 🏀</p>")
@patch("files.helpers.word_censor.SLURS", {'retard': 'r-slur', 'manlet': 'little king', ' nig ': '🏀'})
def test_censor_slurs_does_not_error_out_on_exception():
word_censor.REPLACE_MAP = create_replace_map()
word_censor.REPLACE_MAP["Manlet"] = None
assert_that(censor_slurs(None, ">retarded SuperManlet NIG<")).is_equal_to(">r-slured SuperManlet 🏀<")
@patch("files.helpers.word_censor.SLURS", {'retard': 'r-slur', 'manlet': 'little king'})
def test_censor_slurs_does_not_censor_on_flag_disabled():