def test_delete(test_ds): # given index = SpatialIndexBuilder().set_name("idx1").create_index() index2 = SpatialIndexBuilder().set_name("idx2").create_index() adapter = POINT_TYPE_ADAPTER test_ds.add_type(adapter, index) test_ds.add_type(adapter, index2) write_test_data(test_ds, index, index2) # when qbldr = VectorQueryBuilder() constraints_factory = qbldr.constraints_factory() # filter encompasses 10 features (1, 1) - (10, 10) constraints = constraints_factory.cql_constraints( "BBOX(the_geom, 0.5, 0.5, 10.5, 10.5)") qbldr.constraints(constraints) test_ds.delete(qbldr.build()) query = VectorQueryBuilder().build() res = results_as_list(test_ds.query(query)) # then assert len(test_ds.get_indices()) == 2 assert len(test_ds.get_types()) == 1 assert len(res) == (len(TEST_DATA) - 10)
def test_cql_query(test_ds): # given index = SpatialIndexBuilder().set_name("idx1").create_index() adapter = POINT_TYPE_ADAPTER test_ds.add_type(adapter, index) write_test_data(test_ds, index) # when qbldr = VectorQueryBuilder() constraints_factory = qbldr.constraints_factory() # filter encompasses 10 features (1, 1) - (10, 10) constraints = constraints_factory.cql_constraints( "BBOX(the_geom, 0.5, 0.5, 10.5, 10.5)") qbldr.constraints(constraints) res = results_as_list(test_ds.query(qbldr.build())) # then assert len(res) == 10
def test_query_filter(test_ds): # given index = SpatialIndexBuilder().set_name("idx1").create_index() adapter = POINT_TYPE_ADAPTER test_ds.add_type(adapter, index) write_test_data(test_ds, index) # when qbldr = VectorQueryBuilder() filter_factory = FilterFactory() # filter encompasses 10 features (1, 1) - (10, 10) bbox_filter = filter_factory.bbox(filter_factory.property("the_geom"), 0.5, 0.5, 10.5, 10.5, "EPSG:4326") constraints = qbldr.constraints_factory().filter_constraints(bbox_filter) qbldr.constraints(constraints) res = results_as_list(test_ds.query(qbldr.build())) # then assert len(res) == 10
def test_query_spatial(test_ds): # given index = SpatialIndexBuilder().set_name("idx1").create_index() adapter = POINT_TYPE_ADAPTER test_ds.add_type(adapter, index) write_test_data(test_ds, index) # when qbldr = VectorQueryBuilder() stcb = qbldr.constraints_factory().spatial_temporal_constraints() # polygon encompasses 10 features (1, 1) - (10, 10) stcb.spatial_constraints( Polygon([[0.5, 0.5], [0.5, 10.5], [10.5, 10.5], [10.5, 0.5], [0.5, 0.5]])) stcb.spatial_constraints_compare_operation("CONTAINS") qbldr.constraints(stcb.build()) res = results_as_list(test_ds.query(qbldr.build())) # then assert len(res) == 10
def test_query_temporal(test_ds): # given index = SpatialIndexBuilder().set_name("idx1").create_index() adapter = POINT_TYPE_ADAPTER test_ds.add_type(adapter, index) write_test_data(test_ds, index) # when qbldr = VectorQueryBuilder() stcb = qbldr.constraints_factory().spatial_temporal_constraints() # time range encompasses 10 features (1, 1) - (10, 10) stcb.add_time_range(datetime.fromtimestamp(1), datetime.fromtimestamp(11)) qbldr.constraints(stcb.build()) res = results_as_list(test_ds.query(qbldr.build())) for feature in res: print(feature.get_id()) print(feature.get_default_geometry()) # then assert len(res) == 10
def test_copy_by_query(test_ds): # given options = RocksDBOptions() options.set_geowave_namespace("geowave.tests") options.set_directory(os.path.join(TEST_DIR, "datastore2")) ds2 = DataStoreFactory.create_data_store(options) adapter = POINT_TYPE_ADAPTER index = SpatialIndexBuilder().create_index() test_ds.add_type(adapter, index) write_test_data(test_ds, index) # when qbldr = VectorQueryBuilder() constraints_factory = qbldr.constraints_factory() # filter encompasses 10 features (1, 1) - (10, 10) constraints = constraints_factory.cql_constraints( "BBOX(the_geom, 0.5, 0.5, 10.5, 10.5)") qbldr.all_indices().constraints(constraints) test_ds.copy_to(ds2, qbldr.build()) indices = ds2.get_indices() types = ds2.get_types() query = VectorQueryBuilder().build() res = results_as_list(ds2.query(query)) # then assert len(test_ds.get_indices()) == 1 assert len(indices) == 1 assert indices[0].get_name() == index.get_name() assert indices[0].get_index_strategy() == index.get_index_strategy() assert indices[0].get_index_model() == index.get_index_model() assert len(types) == 1 assert types[0].get_type_name() == adapter.get_type_name() assert len(res) == 10 ds2.delete_all()