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"]]))