test.test_predict

Complexity

Total Complexity

1

Size/Duplication

Total Lines	15
Duplicated Lines	0 %

Importance

Changes

0

1 Function

Rating	Name	Duplication	Size	Complexity
A	test_predict()	0	9	1

import e2edutch.predict

Missing module docstring

from collections import OrderedDict

Unused OrderedDict imported from collections

standard import "from collections import OrderedDict" should be placed before "import e2edutch.predict"

import tensorflow.compat.v1 as tf

Unable to import 'tensorflow.compat.v1'

Unused tensorflow.compat.v1 imported as tf

third party import "import tensorflow.compat.v1 as tf" should be placed before "import e2edutch.predict"

def test_predict():

Missing function or method docstring

    example = {
        'doc_key': 'test',
        'sentences': [['Een', 'zin', '.']]
    }
    predictor = e2edutch.predict.Predictor(model_name='final')
    clusters = predictor.predict(example)
    predictor.end_session()
    assert(len(clusters) >= 0)

There is an unnecessary parenthesis after assert.