예제 #1
0
def test_label_encode_dtype(ncats, cat_dtype):
    s = Series([str(i % ncats) for i in range(ncats + 1)])
    cats = s.unique().astype(s.dtype)
    encoded_col = s.label_encoding(cats=cats)
    np.testing.assert_equal(encoded_col.dtype, cat_dtype)