test.test_bert

Complexity

Total Complexity

1

Size/Duplication

Total Lines	8
Duplicated Lines	0 %

Importance

Changes

0

1 Function

Rating	Name	Duplication	Size	Complexity
A	test_encode_sentences()	0	5	1

import e2edutch.bert

Missing module docstring

def test_encode_sentences():

Missing function or method docstring

    sentences = ['Dit is een zin .'.split(' '), 'Het is slechts een voorbeeld .'.split(' ')]
    tokenizer, model = e2edutch.bert.load_bert('bertje')
    emb = e2edutch.bert.encode_sentences(sentences, tokenizer, model)
    assert emb.shape == (2, 6, 768, 1)