16 lines
623 B
Python
16 lines
623 B
Python
import unittest
|
|
|
|
from src.great_ai.utilities.get_sentences import get_sentences
|
|
|
|
|
|
class TestGetSentences(unittest.TestCase):
|
|
def test_default(self) -> None:
|
|
text = "This is a complete sentence. So is this. However this is n" # ot.
|
|
expected = ["This is a complete sentence.", "So is this.", "However this is n"]
|
|
|
|
self.assertEqual(get_sentences(text), expected)
|
|
self.assertEqual(get_sentences(text, ignore_partial=True), expected[0:2])
|
|
|
|
def test_empty(self) -> None:
|
|
self.assertEqual(get_sentences(""), [])
|
|
self.assertEqual(get_sentences("", ignore_partial=True), [])
|