def test_create_base(self):
"""
Test create base on Lightbase
:return:
"""
status_base = lbstatus.StatusBase()
status_lbbase = status_base.create_base()
self.assertIsInstance(status_lbbase, Base)
result = status_base.remove_base()
self.assertTrue(result)
def __init__(self, name):
"""
Constructor method
:param name: Command name
"""
super(DictionaryCommands, self).__init__(name)
self.dictionary_base = dictionary.DictionaryBase(
dic_base='dictionary'
)
self.status_base = lbstatus.StatusBase(
status_name='status',
dic_name='dictionary'
)
def __init__(self, name):
"""
Constructor method
:param name: Command name
"""
super(AnalyticsCommands, self).__init__(name)
self.status_base = lbstatus.StatusBase(
status_name='status',
dic_name='dictionary'
)
self.analytics_base = analytics.AnalyticsBase(
status_base=self.status_base
)
self.id_doc = None
def test_generate_base(self):
"""
Test creating LB Base from object
:return:
"""
status_base = lbstatus.StatusBase()
lbbase = status_base.lbbase
fd = open('/tmp/status_base.json', 'w+')
fd.write(lbbase.json)
fd.close()
self.assertIsInstance(lbbase, Base)
j = lbbase.json
b = conv.json2base(j)
self.assertIsInstance(b, Base)
def __init__(self, name):
"""
Constructor method
"""
super(TwitterCommands, self).__init__(name)
# Set base to production
self.status_base = lbstatus.StatusBase(status_name='status',
dic_name='dictionary')
self.dictionary_base = dicbase.DictionaryBase(dic_base='dictionary')
self.lbt = lbtwitter.Twitter(debug=False,
term='crime',
status_base=self.status_base,
dictionary_base=self.dictionary_base)
def setUp(self):
"""
Load test data
"""
test_twitter_import.TwitterImportTestCase.setUp(self)
self.lbs = lbsociam.LBSociam()
self.baserest = lbrest.BaseREST(rest_url=self.lbs.lbgenerator_rest_url, response_object=True)
self.lbt = lbtwitter.Twitter(debug=False, term='crime')
self.status_base = lbstatus.StatusBase()
self.tw_status = self.lbt.search()
# Debug
fd = open('/tmp/status_base.json', 'w+')
fd.write(self.status_base.lbbase.json)
fd.close()
# Cria base
self.lbbase = self.status_base.create_base()
pass
def setUp(self):
"""
Ajusta parâmetros iniciais
"""
self.lbs = lbsociam.LBSociam()
self.baserest = lbrest.BaseREST(rest_url=self.lbs.lbgenerator_rest_url,
response_object=True)
self.lbt = lbtwitter.Twitter(debug=False, term='crime')
self.status_base = lbstatus.StatusBase()
self.tw_status = self.lbt.search()
self.data_dir = os.path.join(lbs.lbsociam_data_dir, 'tests')
if not os.path.isdir(self.data_dir):
os.mkdir(self.data_dir)
# Cria base
self.lbbase = self.status_base.create_base()
# Insere dois status
self.status = list()
tw_status_elm = [self.tw_status[0]]
tw_status_json = self.lbt.status_to_json(tw_status_elm)
status_dict = dict(
origin='twitter',
inclusion_date=datetime.datetime.now().strftime("%d/%m/%Y"),
text=tw_status_elm[0].text,
source=tw_status_json,
search_term='crime')
tokenized = srl.srl_tokenize(tw_status_elm[0].text)
if tokenized.get('arg_structures'):
status_dict['arg_structures'] = tokenized.get('arg_structures')
if tokenized.get('tokens'):
status_dict['tokens'] = tokenized.get('tokens')
status = conv.dict2document(self.lbbase, status_dict)
status_json = conv.document2json(self.lbbase, status)
result = self.status_base.documentrest.create(status_json)
self.status.append(status)
# Segundo status
tw_status_elm = [self.tw_status[1]]
tw_status_json = self.lbt.status_to_json(tw_status_elm)
status_dict = dict(
origin='twitter',
inclusion_date=datetime.datetime.now().strftime("%d/%m/%Y"),
text=tw_status_elm[0].text,
source=tw_status_json,
search_term='crime')
tokenized = srl.srl_tokenize(tw_status_elm[0].text)
if tokenized.get('arg_structures'):
status_dict['arg_structures'] = tokenized.get('arg_structures')
if tokenized.get('tokens'):
status_dict['tokens'] = tokenized.get('tokens')
status = conv.dict2document(self.lbbase, status_dict)
status_json = conv.document2json(self.lbbase, status)
result = self.status_base.documentrest.create(status_json)
self.status.append(status)
# Base de dicionário
self.dictionary_base = model_dict.DictionaryBase()