andras/great-ai
1
0
Fork 0
Code Issues Pull requests 1 Projects Releases 4 Packages Wiki Activity Actions

Rename library

Browse source 
Signed-off-by: András Schmelczer <andras@schmelczer.dev>
This commit is contained in:
Andras Schmelczer 2022-05-08 22:04:55 +02:00
parent 8708b157eb
commit eb7ddf1496
No known key found for this signature in database
GPG key ID: 39260B5B0614A13E
141 changed files with 844 additions and 842 deletions
Download patch file Download diff file Expand all files Collapse all files

20
good_ai/tests/sus/test_lemmatize_token.py
View file

@ -1,20 +0,0 @@
import unittest
from src.good_ai.utilities.lemmatize_text import lemmatize_token
from src.good_ai.utilities.nlp import nlp
class TestLemmatizeToken(unittest.TestCase):
def test_simple(self) -> None:
token = nlp("Center")[0]
self.assertEqual(lemmatize_token(token), "centre")
self.assertEqual(lemmatize_token(token, add_negation=True), "centre")
self.assertEqual(lemmatize_token(token, add_part_of_speech=True), "centre_NOUN")
def test_punctuation(self) -> None:
token = nlp("This.")[1]
self.assertEqual(lemmatize_token(token), ".")
self.assertEqual(lemmatize_token(token, add_negation=True), ".")
self.assertEqual(lemmatize_token(token, add_part_of_speech=True), "._PUNCT")