test.test_conll

Complexity

Total Complexity

9

Size/Duplication

Total Lines	63
Duplicated Lines	0 %

Importance

Changes

0

6 Functions

Rating	Name	Duplication	Size	Complexity
A	test_predictions_to_brackets()	0	10	1
A	test_get_doc_key()	0	3	1
A	test_output_conll()	0	11	2
A	test_get_prediction_map()	0	9	1
A	test_output_conll_align()	0	11	3
A	test_get_doc_key_nopart()	0	3	1

import e2edutch.conll

Missing module docstring

import os

standard import "import os" should be placed before "import e2edutch.conll"

__here__ = os.path.dirname(os.path.realpath(__file__))


def test_get_doc_key():

Missing function or method docstring

    doc_key = e2edutch.conll.get_doc_key('test', '1')
    assert doc_key == 'test.p.1'


def test_get_doc_key_nopart():

Missing function or method docstring

    doc_key = e2edutch.conll.get_doc_key('test')
    assert doc_key == 'test'


def test_get_prediction_map():

Missing function or method docstring

    predictions = {'doc1': [[(0, 1)]]}
    prediction_map = e2edutch.conll.get_prediction_map(predictions)
    assert len(prediction_map) == 1
    assert len(prediction_map['doc1']) == 3
    start_map, end_map, word_map = prediction_map['doc1']
    assert start_map[0] == [0]
    assert end_map[1] == [0]
    assert word_map == {}


def test_predictions_to_brackets():

Missing function or method docstring

    sentences = [['Een', 'zin', '.'],
                 ['Nog', 'een', 'zin']]
    predictions = [[(1, 1)], [(4, 5)]]
    brackets = e2edutch.conll.clusters_to_brackets(sentences, predictions)
    assert len(brackets) == 2
    assert len(brackets[0]) == 3
    assert brackets[0][1] == '(0)'
    assert brackets[1][1] == '(1'
    assert brackets[1][2] == '1)'


def test_output_conll():

Missing function or method docstring

    output_file = '/tmp/tmp.conll'
    sentences = {'doc1.p.1': [['Dit', 'is', 'een', 'test', '.']]}
    predictions = {'doc1.p.1': [[(0, 0), (2, 3)]]}
    with open(output_file, 'w') as fout:
        e2edutch.conll.output_conll(fout, sentences, predictions)
    assert os.path.exists(output_file)
    content = open(output_file).readlines()
    nonempty = [line for line in content if line.strip() != '']
    assert len(nonempty) == 7
    assert content[0].strip() == '#begin document (doc1); part 1'


def test_output_conll_align():

Missing function or method docstring

    input_file = os.path.join(__here__, 'data', 'test.conll')
    output_file = '/tmp/tmp.conll'
    predictions = {'doc1.p.1': [[(0, 0), (2, 3)]]}
    with open(input_file) as fin:
        with open(output_file, 'w') as fout:
            e2edutch.conll.output_conll_align(fin, fout, predictions)
    content = open(output_file).readlines()
    nonempty = [line for line in content if line.strip() != '']
    assert len(nonempty) == 7
    assert content[0].strip() == '#begin document (doc1); part 1'