File size: 890 Bytes
59879a5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
import os
import pytest
from geospatial_data_converter.utils import convert, read_file
input_exts = ["kml", "kmz", "geojson", "zip"]
output_exts = ["KML", "ESRI Shapefile", "GeoJSON", "CSV", "OpenFileGDB"]
output_format_dict = {
"ESRI Shapefile": "shp",
"GeoJSON": "geojson",
"CSV": "csv",
"KML": "kml",
"OpenFileGDB": "gdb",
}
@pytest.mark.parametrize("in_ext", input_exts)
@pytest.mark.parametrize("out_ext", output_exts)
def test_kml_coversion(in_ext: str, out_ext: str) -> None:
test_file = f"test.{in_ext}"
test_file_path = os.path.join(os.getcwd(), "tests", "test_data", test_file)
with open(test_file_path, "rb") as f:
kml = read_file(f)
out_file = f"test.{output_format_dict[out_ext]}"
converted_data = convert(kml, out_file, out_ext)
with open("test.kml", "wb") as f:
f.write(converted_data)
|