My playground!
'''
import unittest, os
from analysis.index import Index
from database.warehouse import WarehouseServer
from database.model.tweets import TwoGroupsTweet
BASE_PATH = os.path.expanduser("~/virtualenvfyp/pythia/data/")
index_path = BASE_PATH + "test_index"
ws = WarehouseServer()
sample_docs = ws.get_n_documents(100, type=TwoGroupsTweet)
index = Index(index_path)
for doc in sample_docs:
index.add_document(doc)
index.finalize()
class TestPlayground(unittest.TestCase):
def test_searching(self):
results = index.search_by_term("sales")
calculated = []
for doc in results:
calculated.append(doc.get('id'))
expected = ['4f2d602780286c38a7000013', '4f2d603280286c38a700001e']
self.assertEqual(expected, calculated)
def test_top_terms_index(self):
My playground!
'''
import unittest, os
from analysis.index import Index
from database.warehouse import WarehouseServer
from database.model.tweets import TwoGroupsTweet
BASE_PATH = os.path.expanduser("~/virtualenvfyp/pythia/data/")
index_path = BASE_PATH + "test_index"
ws = WarehouseServer()
sample_docs = ws.get_n_documents(100, type=TwoGroupsTweet)
index = Index(index_path)
for doc in sample_docs:
index.add_document(doc)
index.finalize()
class TestPlayground(unittest.TestCase):
def test_searching(self):
results = index.search_by_term("sales")
calculated = []
for doc in results:
calculated.append(doc.get('id'))
expected = ['4f2d602780286c38a7000013', '4f2d603280286c38a700001e']
self.assertEqual(expected, calculated)
def test_top_terms_index(self):
