wassname/IndicoIo-python
1
0
Fork 0
mirror of https://github.com/wassname/IndicoIo-python.git synced 2026-06-27 16:10:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

ADD: keywords API

Browse Source
This commit is contained in:
jagreene
2015-06-11 09:45:23 -04:00
committed by Madison May
parent 8f4428764e
commit 513ab4101c
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_remote.py
+7 -2
View File
@@ -35,8 +35,10 @@ class BatchAPIRun(unittest.TestCase):
def test_batch_keywords(self): def test_batch_keywords(self):
test_data = ["A working api is key to the success of our young company"] test_data = ["A working api is key to the success of our young company"]
words = [set(text.lower().split()) for text in test_data]
response = batch_keywords(test_data, api_key=self.api_key) response = batch_keywords(test_data, api_key=self.api_key)
self.assertTrue(isinstance(response, list)) self.assertTrue(isinstance(response, list))
self.assertTrue(set(response[0].keys()).issubset(words[0]))
def test_batch_posneg(self): def test_batch_posneg(self):
test_data = ['Worst song ever', 'Best song ever'] test_data = ['Worst song ever', 'Best song ever']
@@ -209,9 +211,13 @@ class FullAPIRun(unittest.TestCase):
def test_keywords(self): def test_keywords(self):
text = "A working api is key to the success of our young company" text = "A working api is key to the success of our young company"
words = set(text.lower().split())
results = keywords(text) results = keywords(text)
assert 'api' in results.keys() sorted_results = sorted(results.keys(), key=lambda x:results.get(x), reverse=True)
assert 'api' in sorted_results[:3]
self.assertTrue(set(results.keys()).issubset(words))
results = keywords(text, top_n=3) results = keywords(text, top_n=3)
assert len(results) is 3 assert len(results) is 3
@@ -220,7 +226,6 @@ class FullAPIRun(unittest.TestCase):
for v in results.values(): for v in results.values():
assert v >= .1 assert v >= .1
def test_political(self): def test_political(self):
political_set = set(['Libertarian', 'Liberal', 'Conservative', 'Green']) political_set = set(['Libertarian', 'Liberal', 'Conservative', 'Green'])
test_string = "Guns don't kill people, people kill people." test_string = "Guns don't kill people, people kill people."