def data_for_sorting(fletcher_type):
"""Length-3 array with a known sort order.
This should be three items [B, C, A] with
A < B < C
"""
return FletcherArray(fletcher_type.data_for_sorting, dtype=fletcher_type.dtype)
def data_for_grouping(fletcher_type):
"""Data for factorization, grouping, and unique tests.
Expected to be like [B, B, NA, NA, A, A, B, C]
Where A < B < C and NA is missing
"""
return FletcherArray(fletcher_type.data_for_grouping, dtype=fletcher_type.dtype)
def data_missing(fletcher_type):
return FletcherArray(fletcher_type.data_missing, dtype=fletcher_type.dtype)
def gen(count):
for _ in range(count):
yield FletcherArray(
fletcher_type.data_repeated(), dtype=fletcher_type.dtype
)
def data(fletcher_type):
return FletcherArray(fletcher_type.data, dtype=fletcher_type.dtype)