def test_nrc_2d(): alphas = np.array([[0.1, 0.2, 0.3, 0.4], [0.4, 0.5, 0.6, 0.7]]) calculated = nrc(alphas) real = np.array([0.25, 0.55]) assert_array_almost_equal(calculated, real)
def test_nrc_1d(): alpha = np.array([0.1, 0.25, 0.5, 0.9]) calculated = nrc(alpha) real = 0.4375 assert_almost_equal(calculated, real)