[IMP] add a few more tests

And refine a defaultdict declaration in AEP.
The previous version was not giving wrong results,
but this one will be more robust.

mis_builder/models/aep.py

@@ -317,7 +317,11 @@ def do_queries(
         aml_model = aml_model.with_context(active_test=False)
         company_rates = self._get_company_rates(date_to)
         # {(domain, mode): {account_id: (debit, credit)}}
-        self._data = defaultdict(lambda: defaultdict(lambda: SimpleArray((0.0, 0.0))))
+        self._data = defaultdict(
+            lambda: defaultdict(
+                lambda: SimpleArray((AccountingNone, AccountingNone)),
+            )
+        )
         domain_by_mode = {}
         ends = []
         for key in self._map_account_ids:

mis_builder/tests/test_aep.py

@@ -90,6 +90,7 @@ def setUp(self):
         self.aep.parse_expr("bali[700IN]")
         self.aep.parse_expr("bale[700IN]")
         self.aep.parse_expr("balp[700IN]")
+        self.aep.parse_expr("balp[700NA]")  # account that does not exist
         self.aep.parse_expr("bali[400AR]")
         self.aep.parse_expr("bale[400AR]")
         self.aep.parse_expr("balp[400AR]")
@@ -200,6 +201,8 @@ def test_aep_basic(self):
         # check ending balance
         self.assertEqual(self._eval("bale[400AR]"), 400)
         self.assertEqual(self._eval("bale[700IN]"), -300)
+        # check result for non existing account
+        self.assertIs(self._eval("bale[700NA]"), AccountingNone)
 
         # let's query for March
         self._do_queries(
@@ -234,9 +237,14 @@ def test_aep_basic(self):
 
         # unallocated p&l from previous year
         self.assertEqual(self._eval("balu[]"), -100)
-
         # TODO allocate profits, and then...
 
+        # let's query for December where there is no data
+        self._do_queries(
+            datetime.date(self.curr_year, 12, 1), datetime.date(self.curr_year, 12, 31)
+        )
+        self.assertIs(self._eval("balp[700IN]"), AccountingNone)
+
     def test_aep_by_account(self):
         self.aep.done_parsing()
         self._do_queries(