def test_label_encoder_encode_to_correct_shape():
encoder = encoders.LabelEncoder(["a", "b"])
dataset = tf.data.Dataset.from_tensor_slices([["a"], ["b"]]).batch(32)
result = encoder.transform(dataset)
assert data_utils.dataset_shape(result).as_list() == [None, 1]
def test_label_encoder_decode_to_same_string():
encoder = encoders.LabelEncoder(["a", "b"])
result = encoder.postprocess([[0], [1]])
assert np.array_equal(result, np.array([["a"], ["b"]]))