From bb8ef22bfd525c67bbee1c5a23677fa461f9c55b Mon Sep 17 00:00:00 2001 From: Jake Howard Date: Mon, 4 Jul 2022 09:26:47 +0100 Subject: [PATCH] Add tests for word count --- website/common/tests.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/website/common/tests.py b/website/common/tests.py index 7bd88b3..296b180 100644 --- a/website/common/tests.py +++ b/website/common/tests.py @@ -2,7 +2,7 @@ from django.test import SimpleTestCase from .embed import YouTubeLiteEmbedFinder from .models import BasePage -from .utils import extract_text, get_page_models, get_table_of_contents +from .utils import count_words, extract_text, get_page_models, get_table_of_contents class BasePageTestCase(SimpleTestCase): @@ -104,3 +104,10 @@ class ExtractTextTestCase(SimpleTestCase): def test_plain_text(self) -> None: self.assertEqual(extract_text("Hello there!"), "Hello there!") + + +class CountWordsTestCase(SimpleTestCase): + def test_counts_words(self) -> None: + self.assertEqual(count_words("a b c"), 3) + self.assertEqual(count_words("Correct Horse Battery Staple"), 4) + self.assertEqual(count_words("Hello there! How are you?"), 5)