import unittest from pathlib import Path import difflib from utils import check_environment_var, setup_logger import plant2dxf class TestDXFGeometry(unittest.TestCase): testordner_path = Path(check_environment_var("PROJECT_TEST")) lib_path = Path(check_environment_var("PROJECT_DATA")) config_dir = Path(check_environment_var("PROJECT_CFG")) work_dir = Path(check_environment_var("PROJECT_WORK")) default_lib_path = lib_path / "blocks.dxf" cfg_path = config_dir / "shapes.cfg" allgemein_cfg_path = config_dir / "allgemein.cfg" GEOM_CODES = {"0", "8", "10", "20", "30", "11", "21", "31", "41", "42", "43", "70", "71", "72", "73", "74", "75"} def extract_geometry_lines(self, file_path): """ Liest eine DXF-Datei ein und gibt nur die geometrischen Zeilen zurück, ignoriert Metadaten wie Timestamps, GUIDs, Layout-Blöcke etc. """ lines = [] with open(file_path, "r", encoding="utf-8") as f: while True: code = f.readline() if not code: break code = code.strip() value = f.readline() if not value: break value = value.strip() if code in self.GEOM_CODES: lines.append(f"{code}\n{value}\n") return lines def assert_dxf_geometry_equal(self, file1, file2): geom1 = self.extract_geometry_lines(file1) geom2 = self.extract_geometry_lines(file2) if geom1 != geom2: diff = "".join(difflib.unified_diff(geom1, geom2, fromfile=str(file1), tofile=str(file2))) raise AssertionError(f"Geometrische Daten unterscheiden sich:\n{diff}") def test_omniflobogen_dxf_file(self): omniflo_bogen = "omniflo_bogen" omniflo_bogen_csv = self.testordner_path / f"{omniflo_bogen}.csv" omniflo_bogen_dxf = f"{omniflo_bogen}.dxf" omniflo_bogen_jason = self.work_dir / f"{omniflo_bogen}.jason" omniflo_bogen_output = self.work_dir / omniflo_bogen_dxf omniflo_bogen_reference = self.testordner_path / omniflo_bogen_dxf plant2dxf.main(omniflo_bogen_csv , self.default_lib_path, self.cfg_path, self.allgemein_cfg_path, omniflo_bogen_output,omniflo_bogen_jason) self.assert_dxf_geometry_equal(omniflo_bogen_output, omniflo_bogen_reference) def test_weiche_90_dxf_file(self): weiche_90= "weiche_90" weiche_90_csv = self.testordner_path / f"{weiche_90}.csv" weiche_90_dxf = f"{weiche_90}.dxf" weiche_90_jason = self.work_dir /f"{weiche_90}.jason" weiche_90_output = self.work_dir / weiche_90_dxf weiche_90_reference = self.testordner_path / weiche_90_dxf plant2dxf.main(weiche_90_csv , self.default_lib_path, self.cfg_path, self.allgemein_cfg_path, weiche_90_output,weiche_90_jason) self.assert_dxf_geometry_equal(weiche_90_output, weiche_90_reference) def test_weiche_45_simple_dxf_file(self): weiche_45_simple= "weiche_45_simple" weiche_45_simple_csv = self.testordner_path / f"{weiche_45_simple}.csv" weiche_45_simple_dxf = f"{weiche_45_simple}.dxf" weiche_45_simple_jason = self.work_dir/ f"{weiche_45_simple}.jason" weiche_45_simple_output = self.work_dir / weiche_45_simple_dxf weiche_45_simple_reference = self.testordner_path / weiche_45_simple_dxf plant2dxf.main(weiche_45_simple_csv , self.default_lib_path, self.cfg_path, self.allgemein_cfg_path, weiche_45_simple_output,weiche_45_simple_jason) self.assert_dxf_geometry_equal(weiche_45_simple_output, weiche_45_simple_reference) def test_weiche_45_doppel_dxf_file(self): weiche_45_doppel= "weiche_45_doppel" weiche_45_doppel_csv = self.testordner_path / f"{weiche_45_doppel}.csv" weiche_45_doppel_dxf = f"{weiche_45_doppel}.dxf" weiche_45_doppel_jason = self.work_dir /f"{weiche_45_doppel}.jason" weiche_45_doppel_output = self.work_dir / weiche_45_doppel_dxf weiche_45_doppel_reference = self.testordner_path / weiche_45_doppel_dxf plant2dxf.main(weiche_45_doppel_csv , self.default_lib_path, self.cfg_path, self.allgemein_cfg_path, weiche_45_doppel_output,weiche_45_doppel_jason) self.assert_dxf_geometry_equal(weiche_45_doppel_output, weiche_45_doppel_reference) def test_weiche_45_dreiwege_dxf_file(self): weiche_45_dreiwege= "weiche_45_dreiwege" weiche_45_dreiwege_csv = self.testordner_path / f"{weiche_45_dreiwege}.csv" weiche_45_dreiwege_dxf = f"{weiche_45_dreiwege}.dxf" weiche_45_dreiwege_jason = self.work_dir /f"{weiche_45_dreiwege}.jason" weiche_45_dreiwege_output = self.work_dir / weiche_45_dreiwege_dxf weiche_45_dreiwege_reference = self.testordner_path / weiche_45_dreiwege_dxf plant2dxf.main(weiche_45_dreiwege_csv , self.default_lib_path, self.cfg_path, self.allgemein_cfg_path, weiche_45_dreiwege_output,weiche_45_dreiwege_jason) self.assert_dxf_geometry_equal(weiche_45_dreiwege_output, weiche_45_dreiwege_reference) def test_weichenkoerper_dxf_file(self): weichenkoerper= "weichenkoerper" weichenkoerper_csv = self.testordner_path / f"{weichenkoerper}.csv" weichenkoerper_dxf = f"{weichenkoerper}.dxf" weichenkoerper_jason = self.work_dir /f"{weichenkoerper}.jason" weichenkoerper_output = self.work_dir / weichenkoerper_dxf weichenkoerper_reference = self.testordner_path / weichenkoerper_dxf plant2dxf.main(weichenkoerper_csv , self.default_lib_path, self.cfg_path, self.allgemein_cfg_path, weichenkoerper_output,weichenkoerper_jason) self.assert_dxf_geometry_equal(weichenkoerper_output, weichenkoerper_reference) def test_weichenkombination_dxf_file(self): weichenkombination= "weichenkombination" weichenkombination_csv = self.testordner_path / f"{weichenkombination}.csv" weichenkombination_dxf = f"{weichenkombination}.dxf" weichenkombination_jason = self.work_dir /f"{weichenkombination}.jason" weichenkombination_output = self.work_dir / weichenkombination_dxf weichenkombination_reference = self.testordner_path / weichenkombination_dxf plant2dxf.main(weichenkombination_csv , self.default_lib_path, self.cfg_path, self.allgemein_cfg_path, weichenkombination_output,weichenkombination_jason) self.assert_dxf_geometry_equal(weichenkombination_output, weichenkombination_reference) def test_gefaelle_dxf_file(self): gefaelle= "gefaelle" gefaelle_csv = self.testordner_path / f"{gefaelle}.csv" gefaelle_dxf = f"{gefaelle}.dxf" gefaelle_jason = self.work_dir /f"{gefaelle}.jason" weichenkombination_output = self.work_dir / gefaelle_dxf weichenkombination_reference = self.testordner_path / gefaelle_dxf plant2dxf.main(gefaelle_csv , self.default_lib_path, self.cfg_path, self.allgemein_cfg_path, weichenkombination_output,gefaelle_jason) self.assert_dxf_geometry_equal(weichenkombination_output, weichenkombination_reference) def test_gefaelle_ausnahme_gleiche_orientierung_dxf_file(self): gefaelle_ausnahme_gleiche_orientierung= "gefaelle_ausnahme_gleiche_orientierung" gefaelle_ausnahme_gleiche_orientierung_csv = self.testordner_path / f"{gefaelle_ausnahme_gleiche_orientierung}.csv" gefaelle_ausnahme_gleiche_orientierung_dxf = f"{gefaelle_ausnahme_gleiche_orientierung}.dxf" gefaelle_ausnahme_gleiche_orientierung_jason = self.work_dir /f"{gefaelle_ausnahme_gleiche_orientierung}.jason" gefaelle_ausnahme_gleiche_orientierung_output = self.work_dir / gefaelle_ausnahme_gleiche_orientierung_dxf gefaelle_ausnahme_gleiche_orientierung_reference = self.testordner_path / gefaelle_ausnahme_gleiche_orientierung_dxf plant2dxf.main(gefaelle_ausnahme_gleiche_orientierung_csv , self.default_lib_path, self.cfg_path, self.allgemein_cfg_path, gefaelle_ausnahme_gleiche_orientierung_output,gefaelle_ausnahme_gleiche_orientierung_jason) self.assert_dxf_geometry_equal(gefaelle_ausnahme_gleiche_orientierung_output, gefaelle_ausnahme_gleiche_orientierung_reference) def test_gefaelle_ausnahme_unterschiedlich_orientierung_dxf_file(self): gefaelle_ausnahme_unterschiedlich_orientierung= "gefaelle_ausnahme_unterschiedlich_orientierung" gefaelle_ausnahme_unterschiedlich_orientierung_csv = self.testordner_path / f"{gefaelle_ausnahme_unterschiedlich_orientierung}.csv" gefaelle_ausnahme_unterschiedlich_orientierung_dxf = f"{gefaelle_ausnahme_unterschiedlich_orientierung}.dxf" gefaelle_ausnahme_unterschiedlich_orientierung_jason = self.work_dir /f"{gefaelle_ausnahme_unterschiedlich_orientierung}.jason" gefaelle_ausnahme_unterschiedlich_orientierung_output = self.work_dir / gefaelle_ausnahme_unterschiedlich_orientierung_dxf gefaelle_ausnahme_unterschiedlich_orientierung_reference = self.testordner_path / gefaelle_ausnahme_unterschiedlich_orientierung_dxf plant2dxf.main(gefaelle_ausnahme_unterschiedlich_orientierung_csv , self.default_lib_path, self.cfg_path, self.allgemein_cfg_path, gefaelle_ausnahme_unterschiedlich_orientierung_output,gefaelle_ausnahme_unterschiedlich_orientierung_jason) self.assert_dxf_geometry_equal(gefaelle_ausnahme_unterschiedlich_orientierung_output, gefaelle_ausnahme_unterschiedlich_orientierung_reference) def test_gefaelle_einzeln_verbunden_dxf_file(self): gefaelle_einzeln_verbunden= "gefaelle_einzeln_verbunden" gefaelle_einzeln_verbunden_csv = self.testordner_path / f"{gefaelle_einzeln_verbunden}.csv" gefaelle_einzeln_verbunden_dxf = f"{gefaelle_einzeln_verbunden}.dxf" gefaelle_einzeln_verbunden_jason = self.work_dir /f"{gefaelle_einzeln_verbunden}.jason" gefaelle_einzeln_verbunden_output = self.work_dir / gefaelle_einzeln_verbunden_dxf gefaelle_einzeln_verbunden_reference = self.testordner_path / gefaelle_einzeln_verbunden_dxf plant2dxf.main(gefaelle_einzeln_verbunden_csv , self.default_lib_path, self.cfg_path, self.allgemein_cfg_path, gefaelle_einzeln_verbunden_output,gefaelle_einzeln_verbunden_jason) self.assert_dxf_geometry_equal(gefaelle_einzeln_verbunden_output, gefaelle_einzeln_verbunden_reference) if __name__ == "__main__": unittest.main()