test_metrics.py 249 B

12345678
  1. from dkma_labor_arbitration.evaluation.metrics import bleu_n, rouge_l
  2. def test_metrics_positive():
  3. pred = "被申请人支付工资"
  4. ref = "被申请人应支付工资"
  5. assert rouge_l(pred, ref) > 0
  6. assert bleu_n(pred, ref, 2) > 0