Adding details to test_delete_arrays

Merge pull request #1119 from openfisca/detaille_tests_delete_arrays
openfisca · Apr 7, 2022 · 97a6543 · 97a6543
2 parents 7103f9a + aba1d53
commit 97a6543
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/core/test_holders.py b/tests/core/test_holders.py
@@ -103,6 +103,12 @@ def test_delete_arrays(single):
     assert simulation.person('salary', '2017-01') == 2500
     assert simulation.person('salary', '2018-01') == 5000
     salary_holder.delete_arrays(period = 2018)
+
+    salary_array = simulation.get_array('salary', '2017-01')
+    assert salary_array is not None
+    salary_array = simulation.get_array('salary', '2018-01')
+    assert salary_array is None
+
     salary_holder.set_input(periods.period(2018), numpy.asarray([15000]))
     assert simulation.person('salary', '2017-01') == 2500
     assert simulation.person('salary', '2018-01') == 1250