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