mirror of
https://github.com/explosion/spaCy.git
synced 2024-11-12 04:38:28 +03:00
31 lines
1.0 KiB
Python
31 lines
1.0 KiB
Python
|
# coding: utf-8
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from ...gold import iob_to_biluo
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize('tags', [('B-ORG', 'L-ORG'),
|
||
|
|
('B-PERSON', 'I-PERSON', 'L-PERSON'),
|
||
|
|
('U-BRAWLER', 'U-BRAWLER')])
|
||
|
|
def test_issue2385_biluo(tags):
|
||
|
|
"""already biluo format"""
|
||
|
|
assert iob_to_biluo(tags) == list(tags)
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize('tags', [('B-BRAWLER', 'I-BRAWLER', 'I-BRAWLER')])
|
||
|
|
def test_issue2385_iob_bcharacter(tags):
|
||
|
|
"""fix bug in labels with a 'b' character"""
|
||
|
|
assert iob_to_biluo(tags) == ['B-BRAWLER', 'I-BRAWLER', 'L-BRAWLER']
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize('tags', [('I-ORG', 'I-ORG', 'B-ORG')])
|
||
|
|
def test_issue2385_iob1(tags):
|
||
|
|
"""maintain support for iob1 format"""
|
||
|
|
assert iob_to_biluo(tags) == ['B-ORG', 'L-ORG', 'U-ORG']
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize('tags', [('B-PERSON', 'I-PERSON', 'B-PERSON')])
|
||
|
|
def test_issue2385_iob2(tags):
|
||
|
|
"""maintain support for iob2 format"""
|
||
|
|
assert iob_to_biluo(tags) == ['B-PERSON', 'L-PERSON', 'U-PERSON']
|