1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
import numpy as np
import pytest
from scipy.stats import random_table
@pytest.mark.parametrize("n", (10, 100, 1000, 10000, 100000))
@pytest.mark.parametrize("k", (2, 4, 10, 20, 40, 100))
@pytest.mark.parametrize("method", (None, "boyett", "patefield"))
def test_rcont(k, n, method, benchmark):
w = np.zeros((k, k))
rng = np.random.default_rng(1)
for _ in range(n):
i = rng.integers(k)
j = rng.integers(k)
w[i, j] += 1
r = np.sum(w, axis=1)
c = np.sum(w, axis=0)
assert np.sum(r) == n
assert np.sum(c) == n
benchmark(lambda: random_table(r, c).rvs(100, method=method, random_state=rng))
|
