Unordered enum data type

[butterlyn]

Unordered enum data type

[stinodego]

If you recommend casting Enums to strings when using polars.testing.assert_frame_equal, perhaps you'd consider reopening this: #16075? 😁 No fuss if not, happy to make do in the meantime since unordered enum can serve the same purpose

No, because the point I made there still stands :)

[mcrumiller]

@butterlyn for now you can just make sure to always sort your categories before creating your Enum dtype.

def sorted_enum(categories):
    return pl.Enum(sorted(categories))

assert_series_equal(
    pl.Series(["a"], dtype=sorted_enum(["a", "b", "c"])),
    pl.Series(["a"], dtype=sorted_enum(["b", "c", "a"])),  # different order
)

[stinodego]

If you just need this for unit tests, you can just write your own assertion util for this, e.g. cast your Enums to strings before checking equality, and then check Enum categories separately.

Still this is an important variant of categorical data that we should support.

[butterlyn]

Thanks @stinodego, yeah I mostly need this for unit testing

If you recommend casting Enums to strings when using polars.testing.assert_frame_equal, perhaps you'd consider reopening this: #16075? 😁 No fuss if not, happy to make do in the meantime since unordered enum can serve the same purpose