fix

openfisca · Oct 29, 2024 · 9007f30 · 9007f30
1 parent a5d44fe
commit 9007f30
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/openfisca_france_data/erfs_fpr/input_data_builder/step_05_foyer.py b/openfisca_france_data/erfs_fpr/input_data_builder/step_05_foyer.py
@@ -41,9 +41,9 @@ def build_foyer(temporary_store = None, year = None):
                                              ),
                                     decls.statut_decl1)
 
-    assert len(couples) == len(decls)
+    assert len(couples[couples.quifam.isin([0,1])]) == len(decls)
     couples = pd.merge(couples,decls[['idfam',"statut_decl1"]],on = 'idfam', how = 'inner')
-    assert len(couples) == len(decls)
+    assert len(couples[couples.quifam.isin([0,1])]) == len(decls)
 
     couples_concubinage = couples[couples['statut_decl1'.isin([3,4,5,6,7,8])]]
     couples_maries = couples[~couples['statut_decl1'.isin([3,4,5,6,7,8])]] ## les couples avec de la non réponses sur le statut marital sont mis en mariés