class ModelTestCase(TestCase):
def setUp(self):
self.test_text = 'Гуляла корова по полю'
self.replaced_text = 'Гуляла ****** по полю'
self.stop_word = Mat(word='коровы')
self.stop_word.save()
def test_replace_field_body(self):
m = TestAntimat(text=self.test_text)
self.assertEqual(m.text, self.replaced_text)
m.text = self.test_text
self.assertEqual(m.text, self.replaced_text)
def test_create(self):
m = TestAntimat.objects.create(text=self.test_text)
self.assertEqual(m.text, self.replaced_text)
m.text = self.test_text
self.assertEqual(m.text, self.replaced_text)
def test_null(self):
for v in [None, False, True]:
m = TestAntimat(text=v)
self.assertEqual(m.text, v)
m = TestAntimat.objects.create(text=v)
self.assertEqual(m.text, v)
class AntimatTestCase(TestCase):
def setUp(self):
self.test_text = 'Гуляла корова по полю'
self.replaced_text = 'Гуляла ****** по полю'
self.stop_word = Mat(word='коровы')
self.stop_word.save()
def test_collect_badword(self):
self.assertEqual(Antimat.collect_badwords(self.test_text), {'КОРОВА'})
def test_replace(self):
self.assertEqual(Antimat.replace_badwords(self.test_text), self.replaced_text)
def test_has_exists_badwords(self):
self.assert_(Antimat.has_exists_badwords(self.test_text))
def setUp(self):
self.test_text = 'Гуляла корова по полю'
self.replaced_text = 'Гуляла ****** по полю'
self.stop_word = Mat(word='коровы')
self.stop_word.save()
