Function test_dist_bures_squared

test/test_gmm.py:94–108

(nx)

@pytest.skip_backend("tf")  # skips because of array assignment
@pytest.skip_backend("jax")
def test_dist_bures_squared(nx):
    m_s, m_t, C_s, C_t, _, _ = get_gmms(nx)
    dist_bures_squared(m_s, m_t, C_s, C_t)
    D0 = dist_bures_squared(m_s, m_s, C_s, C_s)

    assert np.allclose(np.diag(D0), 0, atol=1e-6)

    with pytest.raises(AssertionError):
        dist_bures_squared(m_s[:, 1:], m_t, C_s, C_t)

    with pytest.raises(AssertionError):
        dist_bures_squared(m_s[1:], m_t, C_s, C_t)

    with pytest.raises(AssertionError):
        dist_bures_squared(m_s, m_t[1:], C_s, C_t)


@pytest.skip_backend("tf")  # skips because of array assignment