def search(self, query, max_results=False, fields=None, group_by=None): """ Search the current client for results """ self._wait_for_load() frames = MultiResultList() for item in self: frame = item.search( self._language_parser.parse( query, self._get_item( os.path.basename(item.filename) ).keys() ), max_results=max_results, fields=fields, group_by=group_by ) results = ResultList( name=os.path.basename(item.filename).split('.')[0].split('/')[-1] ) results.dataframe = (frame, self._get_item(os.path.basename(item.filename))) if len(results) > 0: frames.append(results) return frames
def test_result_list_from_pandas_dict(self): dictionary = {0: {'line': 'hello'}, 1: {'line': 'world'}} result_list = ResultList() result_list.dataframe = pd.DataFrame.from_dict(dictionary) self.assertIsInstance(result_list.dataframe, pd.DataFrame)