def test_get_layers(set_schema):
tmin = Variable(tree=[
'select', [
['join', [
{'type': 'Layer', 'id': 26, 'name': 'cnty_24k97', 'field': 'fid'},
{'type': 'Table', 'id': 30, 'name': 'cnty_24k97_data', 'field': 'fid'},
]],
'tmin'
]
])
assert tmin.get_layers() == set([26])
def test_raster(set_schema):
tree = ["select", [
["raster", [
{ "id": "modis_indices_ndvi", "name": "Land indices"},
[ "source", [
{ "field": "fid", "type": "Layer", "id": "26", "name": "cnty_24k97"}
]],
"2015-001,2015-030"
]],
{"field": "mean", "name": "raster"}
]]
mean = Variable(tree=tree)
len(mean.get_layers()) == 1
len(mean.get_rasters()) == 1
