File: test_add_raster.py

package info (click to toggle)
stactools 0.5.3-5
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 23,796 kB
  • sloc: python: 4,498; xml: 554; sh: 395; makefile: 34
file content (27 lines) | stat: -rw-r--r-- 899 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
25
26
27
import pystac
import pystac.utils
from click.testing import CliRunner
from stactools.cli.cli import cli

from tests.conftest import expected_json


def test_add_raster_to_items(tmp_planet_disaster: pystac.Collection):
    collection = tmp_planet_disaster
    collection_path = collection.get_self_href()
    items = list(collection.get_items(recursive=True))
    item_path = pystac.utils.make_absolute_href(
        items[0].get_self_href(), collection_path
    )

    runner = CliRunner()
    result = runner.invoke(cli, ["add-raster", item_path])
    assert result.exit_code == 0

    updated = pystac.read_file(collection_path)
    item = list(updated.get_items(recursive=True))[0]
    asset = item.get_assets().get("analytic")
    assert asset is not None
    expected = expected_json("rasterbands.json")
    for a, b in zip(expected, asset.to_dict().get("raster:bands")):
        assert a == b