def test_build_observed_matrix_simple(): observed = kappa.build_observed_matrix(2, 3, np.array([[0, 1], [1, 1], [0, 0]])) assert observed[0, 0] == 1.0 / 3 assert observed[0, 1] == 1.0 / 3 assert observed[1, 0] == 0 assert observed[1, 1] == 1.0 / 3
def test_build_observed_matrix_longer(): observed = kappa.build_observed_matrix(3, 5, np.array([[0, 1], [2, 2], [1, 1], [0, 2], [1, 1]])) assert observed[0, 0] == 0 assert observed[0, 1] == 1.0 / 5 assert observed[0, 2] == 1.0 / 5 assert observed[1, 0] == 0 assert observed[1, 1] == 2.0 / 5 assert observed[1, 2] == 0 assert observed[2, 0] == 0 assert observed[2, 1] == 0 assert observed[2, 2] == 1.0 / 5
def test_build_observed_matrix_longer(): observed = kappa.build_observed_matrix( 3, 5, np.array([[0, 1], [2, 2], [1, 1], [0, 2], [1, 1]])) assert observed[0, 0] == 0 assert observed[0, 1] == 1.0 / 5 assert observed[0, 2] == 1.0 / 5 assert observed[1, 0] == 0 assert observed[1, 1] == 2.0 / 5 assert observed[1, 2] == 0 assert observed[2, 0] == 0 assert observed[2, 1] == 0 assert observed[2, 2] == 1.0 / 5