mirror of
https://github.com/rasterio/rasterio.git
synced 2025-12-08 17:36:12 +00:00
b954d11106
* renamed variables * improved use of test fixtures * added support for the func's extra args and kwargs * added an example tool
24 lines
792 B
Python
24 lines
792 B
Python
from functools import partial
|
|
|
|
from rasterio.features import dataset_features
|
|
from rasterio import tools
|
|
|
|
|
|
def test_dataset_features_tool(tmpdir, path_rgb_byte_tif):
|
|
"""Example tool works"""
|
|
features_file = tmpdir.join("footprint.jsons")
|
|
|
|
tools.dataset_features_tool(path_rgb_byte_tif, str(features_file), func_kwargs=dict(bidx=1, as_mask=True))
|
|
|
|
assert features_file.read().count("Feature") == 9
|
|
|
|
|
|
def test_dataset_features_partial(tmpdir, path_rgb_byte_tif):
|
|
"""JSON sequence tool works with partially evaluated dataset_features"""
|
|
features_file = tmpdir.join("footprint.jsons")
|
|
tool = tools.JSONSequenceTool(partial(dataset_features, bidx=1, as_mask=True))
|
|
|
|
tool(path_rgb_byte_tif, str(features_file))
|
|
|
|
assert features_file.read().count("Feature") == 9
|