from s3prl.metric import cer, per, wer


def isclose(x: float, y: float) -> float:
    return abs(x - y) < 1e-9


def test_metric():
    # test wer & cer
    hyps = ["a ac abb d"]
    refs = ["a ab abc d"]

    assert isclose(cer(hyps, refs), 0.2)
    assert isclose(wer(hyps, refs), 0.5)
    assert isclose(per(hyps, refs), 0.5)