def test_reproject_rule_layer():
BM = BECModel(TESTCONFIG)
BM.update_config(
{
"rulepolys_file": "tests/data/rulepolys_4326.geojson",
"rulepolys_layer": None
},
reload=True)
assert BM.data["rulepolys"].crs == "EPSG:3005"
def test_load_terraintile_elevation(tmpdir):
BM = BECModel(TESTCONFIG)
BM.update_config(
{
"temp_folder": str(tmpdir),
"rulepolys_file": "tests/data/rulepolys_4326_usa.geojson",
"rulepolys_layer": None
},
reload=True)
BM.load()
assert os.path.exists(tmpdir.join("src", "dem_bc.tif"))
assert os.path.exists(tmpdir.join("src", "dem_exbc.tif"))
assert os.path.exists(tmpdir.join("src", "dem.tif"))
def test_run_shp(tmpdir):
"""
Check that shapefile outputs are created, properly structured and consistent
(not necessarily correct!)
"""
BM = BECModel(TESTCONFIG)
BM.update_config({"temp_folder": str(tmpdir)})
BM.update_config({"out_file": str(os.path.join(tmpdir, "bectest.shp"))})
BM.update_config({"dem": "tests/data/dem_ok.tif"})
BM.load()
BM.model()
BM.postfilter()
BM.write()
assert os.path.exists(tmpdir.join("00_dem.tif"))
assert os.path.exists(tmpdir.join("02_aspect.tif"))
assert os.path.exists(tmpdir.join("bectest.shp"))
assert fiona.listlayers(os.path.join(tmpdir, "bectest.shp")) == ["bectest"]
with fiona.open(os.path.join(tmpdir, "bectest.shp")) as output:
assert list(output.schema["properties"].keys()) == [
"BGC_LABEL",
"AREA_HA",
]
# check outputs
df = gpd.read_file(str(os.path.join(tmpdir, "bectest.shp")))
areas = df.groupby(["BGC_LABEL"])["AREA_HA"].sum().round()
# note output areas are not quite the same as when using gpkg above!
assert list(areas) == [5156.0, 553.0, 3619.0, 7550.0, 1510.0, 5049.0]
def test_nohigh(tmpdir):
BM = BECModel(TESTCONFIG)
BM.update_config({"temp_folder": str(tmpdir)})
BM.update_config(
{"out_file": str(os.path.join(tmpdir, "bectest_nohigh.gpkg"))})
BM.update_config({"elevation": "tests/data/elevation_nohigh.csv"},
reload=True)
BM.load()
BM.model()
BM.postfilter()
BM.write()
def test_load_excel():
BM = BECModel(TESTCONFIG)
BM.update_config({"elevation": "tests/data/elevation.xlsx"})
assert BM.data["elevation"].beclabel[0] == "BG xh 1"
def test_load_becmaster_invalid_data():
with pytest.raises(DataValueError):
BM = BECModel(TESTCONFIG)
BM.update_config(
{"becmaster": "tests/data/becmaster_invalid_data.csv"},
reload=True)
def test_load_becmaster():
BM = BECModel(TESTCONFIG)
BM.update_config({"becmaster": "tests/data/becmaster_test.csv"},
reload=True)
assert BM.data["becmaster"].becvalue[0] == 4
def test_invalid_cell_size3():
with pytest.raises(ConfigValueError):
BM = BECModel(TESTCONFIG)
BM.update_config({"cell_size_metres": 26})
def test_invalid_rule_layer():
with pytest.raises(ConfigValueError):
BM = BECModel(TESTCONFIG)
BM.update_config({"rulepolys_layer": "nodata"})
def test_config_data_missing():
with pytest.raises(ConfigValueError):
BM = BECModel(TESTCONFIG)
BM.update_config({"rulepolys_file": "nodata.gdb"})
def test_local_dem(tmpdir):
"""Test loading dem from local file path
"""
BM = BECModel(TESTCONFIG)
BM.update_config({"dem": "tests/data/dem_ok.tif"})
BM.load()
def test_invalid_dem_path():
with pytest.raises(ConfigValueError):
BM = BECModel(TESTCONFIG)
BM.update_config({"dem": "tests/data/dem_does_not_exit.tif"})
def test_load_invalid_beclabel():
with pytest.raises(DataValueError):
BM = BECModel(TESTCONFIG)
BM.update_config(
{"elevation": "tests/data/elevation_invalid_beclabel.csv"},
reload=True)
def test_load_invalid_rulepolys():
with pytest.raises(DataValueError):
BM = BECModel(TESTCONFIG)
BM.update_config({"rulepolys_file": "tests/data/invalid_data.gdb.zip"},
reload=True)