test(enums): update str eq test (#1267)

openfisca · Oct 9, 2024 · 05e9db7 · 05e9db7
1 parent cfd1749
commit 05e9db7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/openfisca_core/indexed_enums/enum.py b/openfisca_core/indexed_enums/enum.py
@@ -181,8 +181,8 @@ def encode(
 
             >>> array = numpy.array(["TENANT"])
             >>> enum_array = Housing.encode(array)
-            >>> enum_array[0] == Housing.TENANT.index
-            True
+            >>> enum_array == Housing.TENANT
+            array([ True])
 
             # Array of bytes