def test_label_replacement():
    original_labels = ['O', 'location']
    target_labels = ['no_entity', 'location']
    assert (substitute_labels(original_labels, 'O', 'no_entity') == target_labels)
def test_label_replacement():
    original_labels = ['O', 'location']
    target_labels = ['no_entity', 'location']
    assert (substitute_labels(original_labels, 'O', 'no_entity') == target_labels)