| """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 | |