22 lines
583 B
Python
22 lines
583 B
Python
import youtube
|
|
import pytest
|
|
import wordfilter
|
|
|
|
def test_wordfilter():
|
|
wfilter = wordfilter.Wordfilter()
|
|
wfilter.addWords(['FAKE SLUR'])
|
|
|
|
for good in [
|
|
'hi',
|
|
'Nothing in this was made using AI',
|
|
'Nothing in this was made using artificial intelligence',
|
|
]:
|
|
assert youtube.check_text(good, wfilter)
|
|
|
|
for bad in [
|
|
'This music was made using AI',
|
|
'This video was generated using artificial intelligence technology',
|
|
'This is a FAKE SLUR video'
|
|
]:
|
|
assert not youtube.check_text(bad, wfilter)
|