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()