From e0bc5d78333d3668d8a5e338a9a020406562c591 Mon Sep 17 00:00:00 2001
From: cgl <lallemand.chloe@gmail.com>
Date: Fri, 21 Jul 2023 10:44:39 +0200
Subject: [PATCH] fix tests avant 2018

---
 .../input_data_builder/step_06_final.py       | 28 ++++++++++++-------
 1 file changed, 18 insertions(+), 10 deletions(-)

diff --git a/openfisca_france_data/erfs_fpr/input_data_builder/step_06_final.py b/openfisca_france_data/erfs_fpr/input_data_builder/step_06_final.py
index eaa28055..48e06e7c 100644
--- a/openfisca_france_data/erfs_fpr/input_data_builder/step_06_final.py
+++ b/openfisca_france_data/erfs_fpr/input_data_builder/step_06_final.py
@@ -48,16 +48,24 @@ def create_input_data_frame(temporary_store = None, year = None, export_flattene
         "traitement_indiciaire_brut",
         ]
 
-    var_menages = [
-        'idmen',
-        'loyer',
-        'statut_occupation_logement',
-        'taxe_habitation',
-        'wprm',
-        'zone_apl',
-        'logement_conventionne',
-        'prest_precarite_hand' # on récupère la variable de montant de aah / caah pour pouvoir faire une imputation du handicap
-    ]
+    if year >= 2018:
+        var_menages = [
+            'idmen',
+            'loyer',
+            'statut_occupation_logement',
+            'taxe_habitation',
+            'wprm',
+            'zone_apl',
+            'logement_conventionne',
+            'prest_precarite_hand' # on récupère la variable de montant de aah / caah pour pouvoir faire une imputation du handicap
+        ]
+    else:
+        var_menages = [
+            'idmen',
+            'loyer',
+            'taxe_habitation',
+            'wprm',
+            ]
 
     individus = create_ids_and_roles(individus)
     individus = individus[var_individus].copy()