def test_validate():
"""
This method check whether validate function defined in the s2_tiles_supres
file produce the correct results.
"""
test_dir = Path(tempfile.mkdtemp())
valid_desc_10 = [
"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_10, _ = SyntheticImage(20, 18, 4, "uint16",
test_dir).create(seed=45,
transform=transform,
band_desc=valid_desc_10)
validated_10m_indices_exm = [0, 1, 2, 3]
validated_10m_bands_exm = ["B2", "B3", "B4", "B8"]
data_file_path_test = "/test/a.SAFE"
validated_10m_bands, validated_10m_indices, _ = DATA_UTILS(
data_file_path_test).validate(data=test_img_10)
assert set(validated_10m_bands) == set(validated_10m_bands_exm)
assert validated_10m_indices == validated_10m_indices_exm
def test_to_xy():
"""
This method checks to_xy method.
"""
dsr_x_exm = -575834
dsr_y_exm = 66564
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,
32640).create(seed=45,
transform=transform,
band_desc=valid_desc)
# dsr = rasterio.open(test_img)
data_file_path_test = "/test/a.SAFE"
dsr_x, dsr_y = DATA_UTILS(data_file_path_test).to_xy(lon=1,
lat=40,
data=test_img)
assert dsr_x == dsr_x_exm
assert dsr_y == dsr_y_exm
def test_get_max_min():
"""
This method checks the get_min_max method.
"""
dsr_xmin_exm, dsr_ymin_exm, dsr_xmax_exm, dsr_ymax_exm, dsr_area_exm = (
0,
0,
5,
5,
36,
)
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(40, 40, 4, "uint16", test_dir,
32640).create(seed=45,
transform=transform,
band_desc=valid_desc)
# dsr = rasterio.open(test_img)
dsr_xmin, dsr_ymin, dsr_xmax, dsr_ymax, dsr_area = DATA_UTILS.get_max_min(
0, 0, 10, 10, test_img)
assert dsr_xmin == dsr_xmin_exm
assert dsr_ymin == dsr_ymin_exm
assert dsr_xmax == dsr_xmax_exm
assert dsr_ymax == dsr_ymax_exm
assert dsr_area == dsr_area_exm
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)
def test_get_utm():
"""
This method check the get_utm methods.
"""
utm_exm = "epsg:32640"
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(40, 40, 4, "uint16", test_dir,
32640).create(seed=45,
transform=transform,
band_desc=valid_desc)
dsr_utm = DATA_UTILS.get_utm(test_img)
assert dsr_utm == utm_exm
def test_data_final():
"""
This method checks the functionality of data_final method.
"""
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",
]
valid_indices = [0, 1, 2, 3]
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)
d_final = DATA_UTILS.data_final(test_img, valid_indices, 0, 0, 5, 5, 1, 1)
assert d_final.shape == (6, 6, 4)
def test_area_of_interest():
"""
this method checks the area_of_interest methods.
"""
dsr_xmin_exm, dsr_ymin_exm, dsr_xmax_exm, dsr_ymax_exm, dsr_area_exm = (
0,
18,
17,
35,
324,
)
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(40, 40, 4, "uint16", test_dir,
32640).create(seed=45,
transform=transform,
band_desc=valid_desc)
data_file_path_test = "/test/a.SAFE"
clip_to_aoi_test = "75.192123,61.127161,75.195960,61.127993"
dsr_xmin, dsr_ymin, dsr_xmax, dsr_ymax, dsr_area = DATA_UTILS(
data_file_path_test).area_of_interest(test_img, clip_to_aoi_test)
print(dsr_xmin, dsr_ymin, dsr_xmax, dsr_ymax, dsr_area)
assert dsr_xmin == dsr_xmin_exm
assert dsr_ymin == dsr_ymin_exm
assert dsr_xmax == dsr_xmax_exm
assert dsr_ymax == dsr_ymax_exm
assert dsr_area == dsr_area_exm