from image_prediction.label_mapper.mappers.numeric import IndexLabelMapper


def test_index_label_mapper(expected_batch_numeric_labels, expected_batch_string_labels, classes):
    mapper = IndexLabelMapper(classes)
    assert list(mapper(expected_batch_numeric_labels)) == expected_batch_string_labels