From aa7d4519976e1f35d8ee8781c1f4d42e5d90c9ed Mon Sep 17 00:00:00 2001 From: Mahdi Ben Jelloul Date: Mon, 9 Oct 2023 11:55:26 +0100 Subject: [PATCH] Again --- openfisca_survey_manager/tests/test_scenario.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/openfisca_survey_manager/tests/test_scenario.py b/openfisca_survey_manager/tests/test_scenario.py index dbbcade2..c52db107 100644 --- a/openfisca_survey_manager/tests/test_scenario.py +++ b/openfisca_survey_manager/tests/test_scenario.py @@ -351,14 +351,14 @@ def test_inflate(): def test_compute_pivot_table(): survey_scenario = create_randomly_initialized_survey_scenario(collection = None) period = "2017-01" - pivot_table = survey_scenario.compute_pivot_table(columns = ['age'], values = ["salary"], period = period) + pivot_table = survey_scenario.compute_pivot_table(columns = ['age'], values = ["salary"], period = period, simulation = "baseline") assert pivot_table.index == "salary" assert pivot_table.values.round() == 21748 del survey_scenario.weight_variable_by_entity survey_scenario.set_weight_variable_by_entity() - pivot_table = survey_scenario.compute_pivot_table(columns = ['age'], values = ["salary"], period = period) + pivot_table = survey_scenario.compute_pivot_table(columns = ['age'], values = ["salary"], period = period, simulation = "baseline") assert pivot_table.values.round() == 13570. @@ -366,7 +366,7 @@ def test_compute_pivot_table(): def test_compute_quantile(): survey_scenario = create_randomly_initialized_survey_scenario() period = "2017-01" - quintiles = survey_scenario.compute_quantiles(variable = "salary", nquantiles = 5, period = period, weighted = False) + quintiles = survey_scenario.compute_quantiles(variable = "salary", nquantiles = 5, period = period, weighted = False, simulation = "baseline") return quintiles