Display symbolic Enum values in print_computation_log

openfisca_core/indexed_enums.py

@@ -129,3 +129,6 @@ def decode_to_str(self):
 
     def __repr__(self):
         return '{}({})'.format(self.__class__.__name__, str(self.decode()))
+
+    def __str__(self):
+        return str(self.decode_to_str())

tests/core/test_tracer.py

@@ -6,6 +6,7 @@
 from openfisca_core.tracers import Tracer, TracingParameterNodeAtInstant
 from openfisca_core.tools import assert_near
 
+from openfisca_country_template.variables.housing import HousingOccupancyStatus
 from .parameters_fancy_indexing.test_fancy_indexing import parameters
 
 
@@ -40,6 +41,15 @@ def test_log_format():
     assert lines[1] == '    B<2017> >> 1'
 
 
+def test_trace_enums():
+    tracer = Tracer()
+    tracer.record_calculation_start("A", 2017)
+    tracer.record_calculation_end("A", 2017, HousingOccupancyStatus.encode(np.array(['tenant'])))
+
+    lines = tracer.computation_log()
+    assert lines[0] == '  A<2017> >> [\'tenant\']'
+
+
 #  Tests on tracing with fancy indexing
 zone = np.asarray(['z1', 'z2', 'z2', 'z1'])
 housing_occupancy_status = np.asarray(['owner', 'owner', 'tenant', 'tenant'])