Skip to content

Commit b038179

Browse files
committed
refactor(converter): use parametrized tests for vehiculeType translation cases
1 parent 3e0e4ee commit b038179

File tree

1 file changed

+17
-49
lines changed

1 file changed

+17
-49
lines changed

converter/tests/cisu/test_resources_info_converter.py

Lines changed: 17 additions & 49 deletions
Original file line numberDiff line numberDiff line change
@@ -104,55 +104,23 @@ def test_cisu_to_rs_breaking_changes():
104104
validate(rs_message, RS_RI_SCHEMA)
105105

106106

107-
class TestTranslateToCISUVehicleType(TestCase):
108-
def test_translate_SIS(self):
109-
rs_vehicle_type = "SIS"
110-
cisu_vehicle_type = ResourcesInfoCISUConverter.translate_to_cisu_vehicle_type(
111-
rs_vehicle_type
112-
)
113-
self.assertEqual(cisu_vehicle_type, "SIS")
114-
115-
def test_translate_SIS_DRAGON(self):
116-
rs_vehicle_type = "SIS.DRAGON"
117-
cisu_vehicle_type = ResourcesInfoCISUConverter.translate_to_cisu_vehicle_type(
118-
rs_vehicle_type
119-
)
120-
self.assertEqual(cisu_vehicle_type, "SIS")
121-
122-
def test_translate_AUTREVEC(self):
123-
rs_vehicle_type = "AUTREVEC"
124-
cisu_vehicle_type = ResourcesInfoCISUConverter.translate_to_cisu_vehicle_type(
125-
rs_vehicle_type
126-
)
127-
self.assertEqual(cisu_vehicle_type, "AUTRE")
128-
129-
def test_translate_FSI_HELIFSI(self):
130-
rs_vehicle_type = "FSI.HELIFSI"
131-
cisu_vehicle_type = ResourcesInfoCISUConverter.translate_to_cisu_vehicle_type(
132-
rs_vehicle_type
133-
)
134-
self.assertEqual(cisu_vehicle_type, "AUTRE")
135-
136-
def test_translate_SMUR(self):
137-
rs_vehicle_type = "SMUR"
138-
cisu_vehicle_type = ResourcesInfoCISUConverter.translate_to_cisu_vehicle_type(
139-
rs_vehicle_type
140-
)
141-
self.assertEqual(cisu_vehicle_type, "SMUR")
142-
143-
def test_translate_SMUR_VLM(self):
144-
rs_vehicle_type = "SMUR.VLM"
145-
cisu_vehicle_type = ResourcesInfoCISUConverter.translate_to_cisu_vehicle_type(
146-
rs_vehicle_type
147-
)
148-
self.assertEqual(cisu_vehicle_type, "SMUR")
149-
150-
def test_translate_TSU_VSL(self):
151-
rs_vehicle_type = "TSU.VSL"
152-
cisu_vehicle_type = ResourcesInfoCISUConverter.translate_to_cisu_vehicle_type(
153-
rs_vehicle_type
154-
)
155-
self.assertEqual(cisu_vehicle_type, "AUTRE")
107+
@pytest.mark.parametrize(
108+
"rs_vehicule_type,expected",
109+
[
110+
pytest.param("SIS", "SIS", id="translates SIS to SIS"),
111+
pytest.param("SIS.DRAGON", "SIS", id="translates SIS.DRAGON to SIS"),
112+
pytest.param("AUTREVEC", "AUTRE", id="translates AUTREVEC to AUTRE"),
113+
pytest.param("FSI.HELIFSI", "AUTRE", id="translates FSI.HELIFSI to AUTRE"),
114+
pytest.param("SMUR", "SMUR", id="translates SMUR to SMUR"),
115+
pytest.param("SMUR.VLM", "SMUR", id="translates SMUR.VLM to SMUR"),
116+
pytest.param("TSU.VSL", "AUTRE", id="translates TSU.VSL to AUTRE"),
117+
],
118+
)
119+
def test_translate_vehicule_type_to_cisu(rs_vehicule_type, expected):
120+
cisu_vehicle_type = ResourcesInfoCISUConverter.translate_to_cisu_vehicle_type(
121+
rs_vehicule_type
122+
)
123+
assert cisu_vehicle_type == expected
156124

157125

158126
@pytest.mark.parametrize(

0 commit comments

Comments
 (0)