tests/test_gen_cmat_old.py

"""Test old gen_cmat."""
from loadtext import loadtext
from st_mlbee.gen_cmat import gen_cmat


def test_gen_cmat_old():
    """Test old gen_cam."""
    paras1 = loadtext("data/sternstunden04-en.txt", splitlines=True)
    paras2 = loadtext("data/sternstunden04-de.txt", True)

    cmat = gen_cmat(paras1, paras2)
    len1, len2 = len(paras1), len(paras2)

    # note the order
    assert cmat.shape == (len2, len1)


if __name__ == "__main__":
    test_gen_cmat_old()