def test_vectors_aggregate_query_complex(self): wkt = 'POLYGON((-76.65 40.10, -76.65 40.14, -76.55 40.14, -76.55 40.10, -76.65 40.10))' child_agg = AggregationDef(agg_type='date_hist', value='M') aggs = AggregationDef(agg_type='geohash', value='4', children=child_agg) v = Vectors() query = 'item_type:tweet' start_date = 'now-6M' end_date = 'now' result = v.aggregate_query(wkt, aggs, index=None, query=query, start_date=start_date, end_date=end_date) assert len(result) == 1 assert 'name' in result[0] assert result[0]['name'] == 'geohash:4' assert 'terms' in result[0] terms = result[0]['terms'] assert len(terms) == 1 assert terms[0]['term'] == 'dr1s' assert len(terms[0]['aggregations']) == 1 assert len(terms[0]['aggregations'][0]['terms']) == 4
def test_vectors_aggregate_query_agg_def(self): wkt = 'POLYGON((-76.65 40.10, -76.65 40.14, -76.55 40.14, -76.55 40.10, -76.65 40.10))' aggs = AggregationDef(agg_type='terms', value='ingest_source') v = Vectors() result = v.aggregate_query(wkt, aggs, index=None) assert len(result) == 1 assert 'name' in result[0] assert result[0]['name'] == 'terms:ingest_source' assert 'terms' in result[0] assert len(result[0]['terms']) == 6
def test_vectors_aggregate_query_with_defined_index(self): wkt = 'POLYGON((-76.65 40.10, -76.65 40.14, -76.55 40.14, -76.55 40.10, -76.65 40.10))' aggs = 'terms:item_type' v = Vectors() result = v.aggregate_query(wkt, aggs, index='read-vector-osm-*') assert len(result) == 1 assert 'name' in result[0] assert result[0]['name'] == 'terms:item_type' assert 'terms' in result[0] assert len(result[0]['terms']) == 10