tests/test_gradio_cmat.py

"""Test gradio_cmat."""
from gradio_cmat import gradio_cmat


def test_gradio_cmat():
    """Test gradio_cmat."""
    list1 = ["test", "测试", "love"]
    list2 = ["this is a test"]
    cmat = gradio_cmat(list1, list2)
    assert cmat.shape == (3, 1)

    assert cmat[0, 0] > 0.6
    assert cmat[1, 0] > 0.6
    assert cmat[2, 0] < 0.2

    # cmat.round(2)