def test_search_by_image_url_and_geo_box(channel):
stub = service_pb2_grpc.V2Stub(channel)
with SetupImage(stub) as input_:
response = stub.PostSearches(
service_pb2.PostSearchesRequest(
query=resources_pb2.Query(ands=[
resources_pb2.And(output=resources_pb2.Output(
input=resources_pb2.Input(data=resources_pb2.Data(
image=resources_pb2.Image(url=DOG_IMAGE_URL))))),
resources_pb2.And(input=resources_pb2.Input(
data=resources_pb2.Data(geo=resources_pb2.Geo(geo_box=[
resources_pb2.GeoBoxedPoint(
geo_point=resources_pb2.GeoPoint(longitude=43,
latitude=54)),
resources_pb2.GeoBoxedPoint(
geo_point=resources_pb2.GeoPoint(longitude=45,
latitude=56)),
])))),
]),
pagination=service_pb2.Pagination(page=1, per_page=1000),
),
metadata=metadata(),
)
raise_on_failure(response)
assert len(response.hits) > 0
assert input_.id in [hit.input.id for hit in response.hits]
def test_search_by_geo_box_and_annotated_name_and_predicted_name(channel):
stub = service_pb2_grpc.V2Stub(channel)
with SetupImage(stub) as input_:
my_concept_name = input_.data.concepts[0].name
response = stub.PostSearches(
service_pb2.PostSearchesRequest(
query=resources_pb2.Query(ands=[
resources_pb2.And(input=resources_pb2.Input(
data=resources_pb2.Data(geo=resources_pb2.Geo(geo_box=[
resources_pb2.GeoBoxedPoint(
geo_point=resources_pb2.GeoPoint(longitude=43,
latitude=54)),
resources_pb2.GeoBoxedPoint(
geo_point=resources_pb2.GeoPoint(longitude=45,
latitude=56)),
])))),
resources_pb2.And(input=resources_pb2.Input(
data=resources_pb2.Data(concepts=[
resources_pb2.Concept(name=my_concept_name,
value=1)
]))),
resources_pb2.And(output=resources_pb2.Output(
data=resources_pb2.Data(concepts=[
resources_pb2.Concept(name="dog", value=1)
]))),
]),
pagination=service_pb2.Pagination(page=1, per_page=1000),
),
metadata=metadata(),
)
raise_on_failure(response)
assert len(response.hits) > 0
assert input_.id in [hit.input.id for hit in response.hits]
def test_search_by_image_url(channel):
stub = service_pb2_grpc.V2Stub(channel)
with SetupImage(stub) as input_:
response = stub.PostSearches(
service_pb2.PostSearchesRequest(
query=resources_pb2.Query(ands=[
resources_pb2.And(output=resources_pb2.Output(
input=resources_pb2.Input(data=resources_pb2.Data(
image=resources_pb2.Image(url=DOG_IMAGE_URL)))))
]),
pagination=service_pb2.Pagination(page=1, per_page=1000),
),
metadata=metadata(),
)
raise_on_failure(response)
assert len(response.hits) > 0
assert input_.id in [hit.input.id for hit in response.hits]
def test_search_by_predicted_concept_name(channel):
stub = service_pb2_grpc.V2Stub(channel)
with SetupImage(stub) as input_:
response = stub.PostSearches(
service_pb2.PostSearchesRequest(
query=resources_pb2.Query(ands=[
resources_pb2.And(output=resources_pb2.Output(
data=resources_pb2.Data(concepts=[
resources_pb2.Concept(name="dog", value=1)
])))
]),
pagination=service_pb2.Pagination(page=1, per_page=1000),
),
metadata=metadata(),
)
raise_on_failure(response)
assert len(response.hits) > 0
assert input_.id in [hit.input.id for hit in response.hits]
