def test_censor6():
frase = "HoLa FlOjO BoLuDo"
cosine = CosineSimilarity(threshold=0.3, normalize=False)
palabrota = Palabrota(countries=[Country.COLOMBIA, Country.VENEZUELA], censor_char='-',
exclude=[], include=[], distance_metric=cosine)
result = palabrota.censor(frase)
assert result != frase
def test_censor2():
frase = "hola huevon flojo"
palabrota = Palabrota(countries=[Country.VENEZUELA], censor_char='.', exclude=["huevon"], include=[],
distance_metric=None)
result = palabrota.censor(frase)
assert result == "hola huevon ....."
assert len(result) == len(frase)
def test_supported_countries():
palabrota = Palabrota()
assert len(palabrota.supported_countries()) > 0
def test_contains_palabrota1():
palabrota = Palabrota(countries=[Country.COLOMBIA])
assert palabrota.contains_palabrota("huevon")
assert not palabrota.contains_palabrota("")
def test_test_levenshtein_distance3():
palabrota = Palabrota(censor_char="*", countries=[Country.VENEZUELA], distance_metric=LevenshteinDistance())
print(palabrota.censor("Huevon", "huevon"))
def test_hamming_distance3():
palabrota = Palabrota(censor_char="*", countries=[Country.VENEZUELA], distance_metric=HammingDistance())
print(palabrota.censor("Huevon", "huevon"))
def test_censor4():
frase = "hola flojo boludo"
palabrota = Palabrota(countries=[Country.VENEZUELA], censor_char='-', exclude=["hola"], include=["boludo"],
distance_metric=None)
result = palabrota.censor(frase)
assert result == "hola ----- ------"
def test_censor1():
frase = "hola huevon"
palabrota = Palabrota(countries=[Country.COLOMBIA], censor_char='*', exclude=[], include=[], distance_metric=None)
result = palabrota.censor(frase)
assert result == "hola ******"
assert len(result) == len(frase)