Skip to content

Commit 8733539

Browse files
committed
Fix test error
1 parent 4b90c20 commit 8733539

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/test_text_mining.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,13 +2,16 @@
22
from unittest import TestCase
33
import numpy as np
44
import pandas as pd
5+
import nltk
56

67
from niaarm.niaarm import _cut_point
78
from niaarm.text import Corpus, TextRule, NiaARTM
89

910

1011
class TestTextMining(TestCase):
1112
def setUp(self):
13+
nltk.download('punkt')
14+
nltk.download('stopwords')
1215
ds_path = os.path.join(os.path.dirname(__file__), 'test_data', 'artm_test_dataset.json')
1316
df = pd.read_json(ds_path, orient='records')
1417
documents = df['text'].tolist()

0 commit comments

Comments
 (0)