0d0b86f0c15a40b566029a3d24123ea1f48eee8c,dit/math/tests/test_pmfops.py,,test_clamps,#,125

Before Change

def test_clamps():
    d = np.array([.51, .48, .01])
    out_ = (np.array([[4, 3, 0], [5, 4, 1]]),
            np.array([ 0., 0.125, 0.25, 0.375, 0.5, 0.625, 0.75, 0.875, 1.]))
    out = module.clamped_indexes(d, 2**3)
    np.testing.assert_allclose(out[0], out_[0])
    np.testing.assert_allclose(out[1], out_[1], rtol=1e-7, atol=1e-8)

After Change

def test_clamps():
    d = np.array([.51, .48, .01])
    locs = np.linspace(0, 1, 2**3+1)
    out_ = np.array([[4, 3, 0], [5, 4, 1]])
    out = module.clamped_indexes(d, locs)
    np.testing.assert_allclose(out, out_)

Instances