def test():
spans = generate_spans()
iob = generate_iob(spans)
gold_bio = generate_bio(spans)
bio = convert_iob_to_bio(iob)
assert bio == gold_bio
def test_iob_bio_i_first():
in_ = ["I-X", "O"]
gold = ["B-X", "O"]
res = convert_iob_to_bio(in_)
assert res == gold
def test_iob_bio_i_after_b_diff():
in_ = ["O", "B-X", "I-Y", "O"]
gold = ["O", "B-X", "B-Y", "O"]
res = convert_iob_to_bio(in_)
assert res == gold
def test_iob_bio_i_after_i_same():
in_ = ["O", "I-X", "I-X", "O"]
gold = ["O", "B-X", "I-X", "O"]
res = convert_iob_to_bio(in_)
assert res == gold
def test():
spans = generate_spans()
gold_bio = generate_bio(spans)
res = convert_iob_to_bio(convert_bio_to_iob(gold_bio))
assert res == gold_bio