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

openfisca · Oct 9, 2024 · 5259224 · 5259224
1 parent 8074af4
commit 5259224
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