File: test_query.py

package info (click to toggle)
stac-pydantic 3.4.0-1
  • links: PTS, VCS
  • area: main
  • in suites:
  • size: 664 kB
  • sloc: python: 1,787; makefile: 4
file content (24 lines) | stat: -rw-r--r-- 645 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
import pytest
from pydantic import ValidationError

from stac_pydantic.api.search import ExtendedSearch, Search


def test_api_query_extension():
    # One field
    Search(collections=["collection1", "collection2"], query={"field": {"lt": 100}})

    # Many fields
    Search(
        collections=["collection1", "collection2"],
        query={"field": {"lt": 100}, "field1": {"gt": 200}},
    )


def test_api_query_extension_invalid():
    # Invalid operator
    with pytest.raises(ValidationError):
        ExtendedSearch(
            collections=["collection1", "collection2"],
            query={"field": {"greater_than": 100}},
        )