Skip to content

Commit

Permalink
Added test case to get all keywords from kp
Browse files Browse the repository at this point in the history
  • Loading branch information
vi3k6i5 committed Nov 19, 2017
1 parent b9e7e15 commit 86f01ef
Showing 1 changed file with 29 additions and 0 deletions.
29 changes: 29 additions & 0 deletions test/test_kp_get_all_keywords.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,29 @@
from collections import defaultdict
from flashtext import KeywordProcessor
import logging
import unittest
import json
import re

logger = logging.getLogger(__name__)


class TestKPKeywords(unittest.TestCase):
def setUp(self):
logger.info("Starting...")

def tearDown(self):
logger.info("Ending.")

def test_list_loading(self):
keyword_processor = KeywordProcessor()
keyword_processor.add_keyword('j2ee', 'Java')
keyword_processor.add_keyword('onGoing', 'rendom')
keyword_processor.get_all_keywords()
self.assertEqual(keyword_processor.get_all_keywords(),
{'j2ee': 'Java', 'ongoing': 'rendom'},
"get_all_keywords didn't match expected results.")


if __name__ == '__main__':
unittest.main()

0 comments on commit 86f01ef

Please sign in to comment.