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