Add support for optional arguments to API

This commit is contained in:
Madison May
2014-12-18 16:10:36 -05:00
parent 8b9e0841a0
commit 6c23d0d7ee
11 changed files with 35 additions and 34 deletions
+5
View File
@@ -27,6 +27,11 @@ class FullAPIRun(unittest.TestCase):
results = text_tags(text)
max_keys = sorted(results.keys(), key=lambda x:results.get(x), reverse=True)
assert 'political_discussion' in max_keys[:5]
results = text_tags(text, top_n=5)
assert len(results) is 5
results = text_tags(text, threshold=0.1)
for v in results.values():
assert v >= 0.1
def test_political(self):
political_set = set(['Libertarian', 'Liberal', 'Conservative', 'Green'])