Python ModelStore.handle_resolution_overlap Beispiele

Beispiel #1

Datei: test_models.py Projekt: karansingla06/ICE-NER-NLP

def test_handle_resolution_overlap_with_overlap():
    store = ModelStore()
    resp = store.handle_resolution_overlap([{
        'start': 0,
        'end': 2,
        'entity': 'karan singla'
    }, {
        'start': 1,
        'end': 3,
        'entity': 'singla is'
    }, {
        'start': 2,
        'end': 6,
        'entity': 'is famous in Delhi'
    }])
    expected = [{
        'start': 0,
        'end': 2,
        'entity': 'karan singla'
    }, {
        'start': 2,
        'end': 6,
        'entity': 'is famous in Delhi'
    }]
    assert resp == expected

Beispiel #2

Datei: test_models.py Projekt: karansingla06/ICE-NER-NLP

def test_handle_resolution_overlap_without_overlap():
    store = ModelStore()
    resp = store.handle_resolution_overlap([{
        'start': 4,
        'end': 6
    }, {
        'start': 6,
        'end': 7
    }, {
        'start': 8,
        'end': 10
    }])
    expected = [{
        'start': 4,
        'end': 6
    }, {
        'start': 6,
        'end': 7
    }, {
        'start': 8,
        'end': 10
    }]
    assert resp == expected

Beispiel #3

Datei: test_models.py Projekt: karansingla06/ICE-NER-NLP

def test_handle_resolution_overlap_with_empty_list():
    store = ModelStore()
    resp = store.handle_resolution_overlap([])
    expected = []
    assert resp == expected