|
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)
|
|
|
