tests/test_conversions.py

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)