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)