def test(): """This test is used for sample dataset""" db = get_db("sample_dataset") data = load_data("sample.osm.json" ) stats = get_query(data, db) pipelines = get_pipeline() result = db.sample.aggregate(pipelines[1]) assert stats == {} assert result == {u'ok': 1.0, u'result': []} print "Pass tests"
def output_result(): """This function print output for queries""" db = get_db("las_vegas_map_dataset_R1") data = load_data("las-vegas_nevada.osm.json") stats = get_query(data, db) #print number of each type of amenity print stats pipelines = get_pipeline() #describ = ["Hotels as amenity: ", "Hotels as tourism: ", "Malls as amenity: ", "Malls as shop: ", "Top 5 contributors to the map: "] #query before database update (database:las_vegas_map_dataset) describ = ["Hotels as amenity: ", "Malls as amenity: ", "Top 5 contributors to the map: "] #query after database update (database:las_vegas_map_dataset_R1) for index in range(len(pipelines)): result = db.las_vegas_map_dataset_R1.aggregate(pipelines[index]) print describ[index], len(result[u'result']) pprint.pprint(result) #print result of each pipeline print "\n"