def test_simple_search(self):
try:
search_index = SearchIndex(self.context)
hits, searcher = search_index.search('test')
for i, doc in hits:
foo, bar = doc.get('uid'), hits.score(i)
if searcher:
searcher.close()
except Exception, e:
self.fail('Error searching: %s' % e)
config.read_file(options.config)
context = canary.context.Context(config)
source_catalog = context.get_source_catalog()
ctm = source_catalog.get_complete_mapping()
if not args:
print usage
sys.exit(0)
query_str = options.boolean.join(
[' "%s" [%s] ' % (term, options.field) for term in args])
#print query_str.strip()
search_index = SearchIndex(context)
hit_list = []
hits, searcher = search_index.search(query_str)
for i, doc in hits:
hit_list.append(doc.get('uid'))
searcher.close()
output = []
for id in hit_list:
rec = QueuedRecord(context, int(id))
if options.locations:
study = Study(context, rec.study_id)
for loc in study.locations:
out = []
out.extend((id, loc.uid, loc.study_id, loc.feature_id))
feature = Feature(uid=loc.feature_id)
feature.load(context)
out.extend((feature.latitude, feature.longitude, feature.name,
search_index = SearchIndex(context)
searches = (
'Lead [exposure]',
'Hantavirus [exposure]',
'peromyscus [species]',
'Michigan [location]',
'DDT [exposure]',
'2003 [year]',
'cats and dogs',
'"Burger J" [author]',
'cross-sectional [methodology]',
'case-control [methodology] and cattle [species]',
'disease-model [methodology]',
'"age factors" [risk_factors]',
'Sarin [exposure]',
'Arsenic [exposure]',
'"Bacillus anthracis" [exposure]',
'"West Nile Virus" [exposure]',
'"Water Pollutants, Chemical" [exposure]',
)
for t in searches:
hits, searcher = search_index.search(t)
result_set = []
for i, doc in hits:
uid = doc.get(str('uid'))
record = QueuedRecord(context, uid)
if record.status == record.STATUS_CURATED:
result_set.append(record)
searcher.close()
makeplot(context, t, result_set)
