test_misc.py 258 Bytes
Newer Older
Leo Gao's avatar
Leo Gao committed
1
2
3
4
5
6
7
8
9
10
import pytest
import lm_eval.metrics as metrics


def test_bootstrapping():
    arr = list(range(100))
    expected = metrics.mean_stderr(arr)
    bootstrapped = metrics.bootstrap_stderr(metrics.mean, arr)

    assert bootstrapped == pytest.approx(expected)