Plaisance voile légère en loisir nautique #95

entrepreneur-interet-general · Oct 6, 2018 · c6e1574 · c6e1574
1 parent 943a214
commit c6e1574
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/opendata/tests/files/expected_flotteurs.csv b/opendata/tests/files/expected_flotteurs.csv
@@ -4,6 +4,7 @@ operation_id,numero_ordre,pavillon,resultat_flotteur,type_flotteur,categorie_flo
 -40995,1,Français,Assisté,Plaisance à moteur < 8m,Plaisance,b12ab00e47a8914f32bf7bacf87792a0acdf4f35,,EDEL-5,True,5.4,0,2.33,0,,14.9,6.62,MONOCOQUE NON HABITABLE,POLYESTER / EPOXY,Voile,HORS-BORD,VOILIER,USAGE PERSONNEL
 -39436,2,Étranger,Assisté,Plaisance à voile,Plaisance,,,,,,,,,,,,,,,,,
 2219940324,1,Français,Remorqué,Pêche,Pêche,,,,,,,,,,,,,,,,,
+2219940325,1,Français,Remorqué,Plaisance voile légère,Loisir nautique,,,,,,,,,,,,,,,,,
 2220013242,1,Français,Assisté,Plaisance à voile,Plaisance,1b9161998ffaecec2285a66d583f7c1ece2d04d1,,FIRST-29,True,8.75,0,7.37,0,,38,9.57,MONOCOQUE NON HABITABLE,POLYESTER / EPOXY,Voile,LIGNE D'ARBRE,INCONNU,USAGE PERSONNEL
 2220070718,1,Français,Remorqué,Plaisance à voile,Plaisance,c1f96c096b5c97acf04817ab6a94bf93581f96d5,BENETEAU,OCEANIS CLIPPER 343,True,10.38,3.48,12.65,0,30,50.5,21.34,MONOCOQUE HABITABLE,POLYESTER / EPOXY,Voile,LIGNE D'ARBRE,VOILIER,USAGE PERSONNEL
 2319910129,1,,"Difficulté surmontée, reprise de route",Planche à voile,Loisir nautique,,,,,,,,,,,,,,,,,

diff --git a/opendata/tests/files/flotteurs.csv b/opendata/tests/files/flotteurs.csv
@@ -8,3 +8,4 @@ operation_id,numero_ordre,pavillon,resultat_flotteur,type_flotteur,categorie_flo
 -40995,1,Français,Assisté,Plaisance à moteur < 8m,Plaisance,387047,,EDEL-5,1,5.40,0.00,2.33,0,,14.90,6.62,MONOCOQUE NON HABITABLE,POLYESTER / EPOXY,Voile,HORS-BORD,VOILIER,USAGE PERSONNEL
 2220070718,1,Français,Remorqué,Plaisance à voile,Plaisance,C44335,BENETEAU,OCEANIS CLIPPER 343,1,10.38,3.48,12.65,0,30.00,50.50,21.34,MONOCOQUE HABITABLE,POLYESTER / EPOXY,Voile,LIGNE D'ARBRE,VOILIER,USAGE PERSONNEL
 2219940324,1,Français,Remorqué,Pêche,Pêche,,,,,,,,,,,,,,,,,
+2219940325,1,Français,Remorqué,Plaisance voile légère,Plaisance,,,,,,,,,,,,,,,,,
diff --git a/opendata/transformers/flotteurs.py b/opendata/transformers/flotteurs.py
@@ -13,8 +13,12 @@ def __init__(self, filepath):
 
     def transform(self, output):
         df = self.read_csv()
+
         df['numero_immatriculation'] = self.build_numero_immatriculation(df)
         df['assurance'] = self.assurance(df.assurance)
+        plaisance_voile_legere = df.type_flotteur == 'Plaisance voile légère', 'categorie_flotteur'
+        df.loc[plaisance_voile_legere] = 'Loisir nautique'
+
         self.to_csv(df, output)
 
     def assurance(self, series):