mlbee / tests1 /test_gen_cmat_old.py
ffreemt
Replace model-s-cpu with load_model_s (model-s-512 model-s-512-v2 on hf)
34815a7
raw
history blame
468 Bytes
"""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()