File size: 556 Bytes
2ff3279 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
"""Test debee."""
# pylint: disable=broad-except
from debee.gen_cmat import gen_cmat
from debee.loadparas import loadparas
def test_gen_cmat():
"""Test gen_cmat."""
text1 = loadparas("data/sternstunden04-de.txt")
text2 = loadparas("data/sternstunden04-en.txt")
assert len(text1) >= 29
assert len(text2) >= 30
cmat = gen_cmat(text1, text2)
cmat0 = gen_cmat(text1, text2, remove_punctuation=False)
len_21 = (len(text2), len(text1))
assert cmat.shape == len_21
assert cmat.sum() > 59
assert cmat0.sum() > 59.1
|