def test_get_band_short_name():
"""
This method checks the functionality of get_short_name methods.
"""
short_desc_exm = ["B4", "B3", "B2", "B8"]
test_dir = Path(tempfile.mkdtemp())
valid_desc = [
"B4, central wavelength 665 nm",
"B3, central wavelength 560 nm",
"B2, central wavelength 490 nm",
"B8, central wavelength 842 nm",
]
transform = from_origin(1470996, 6914001, 10.0, 10.0)
test_img, _ = SyntheticImage(20, 18, 4, "uint16",
test_dir).create(seed=45,
transform=transform,
band_desc=valid_desc)
dsr = rasterio.open(test_img)
short_desc = []
for i in range(dsr.count):
desc = DATA_UTILS.validate_description(dsr.descriptions[i])
short_desc.append(DATA_UTILS.get_band_short_name(desc))
assert set(short_desc) == set(short_desc_exm)
def test_validate_description():
"""
this method checks the validate_description methods.
"""
valid_desc_exm = [
"B4 (665 nm)", "B3 (560 nm)", "B2 (490 nm)", "B8 (842 nm)"
]
test_dir = Path(tempfile.mkdtemp())
valid_desc = [
"B4, central wavelength 665 nm",
"B3, central wavelength 560 nm",
"B2, central wavelength 490 nm",
"B8, central wavelength 842 nm",
]
transform = from_origin(1470996, 6914001, 10.0, 10.0)
test_img, _ = SyntheticImage(20, 18, 4, "uint16",
test_dir).create(seed=45,
transform=transform,
band_desc=valid_desc)
dsr = rasterio.open(test_img)
valid_desc = []
print(dsr.count)
for i in range(dsr.count):
valid_desc.append(DATA_UTILS.validate_description(dsr.descriptions[i]))
assert set(valid_desc) == set(valid_desc_exm)