示例#1
0
def test_basic_redux():
    x = np.linspace(-100, 100, 1000)
    y = 10*np.exp(-(x**2/(2*20**2)))

    als = AlsCvxopt(smoothness_param=1e-2, asym_param=1e-1, 
                    redux=10, fix_end_points=False, fix_rng=None, 
                    verbose=True)

    y_als = als.calculate(y)

    np.testing.assert_almost_equal(y, y_als, decimal=3)
示例#2
0
def test_basic_redux():
    x = np.linspace(-100, 100, 1000)
    y = 10 * np.exp(-(x**2 / (2 * 20**2)))

    als = AlsCvxopt(smoothness_param=1e-2,
                    asym_param=1e-1,
                    redux=10,
                    fix_end_points=False,
                    fix_rng=None,
                    verbose=True)

    y_als = als.calculate(y)

    np.testing.assert_almost_equal(y, y_als, decimal=3)
示例#3
0
def test_basic_rng():
    x = np.linspace(-100, 100, 1000)
    y = 10*np.exp(-(x**2/(2*20**2)))

    rng = np.arange(200,800)
    als = AlsCvxopt(smoothness_param=1, asym_param=1e-3, rng=rng,
                    redux=1, fix_end_points=False, fix_rng=None, 
                    verbose=True)

    y_als = als.calculate(y)

    np.testing.assert_almost_equal(y[...,rng], y_als[...,rng], decimal=2)
    np.testing.assert_allclose(y_als[:200],0)
    np.testing.assert_allclose(y_als[800:],0)
示例#4
0
def test_fix_rng():
    x = np.linspace(-100, 100, 1000)
    y = 10*np.exp(-(x**2/(2*20**2)))
    fix_rng = np.arange(200,800)

    als = AlsCvxopt(smoothness_param=1e3, asym_param=1e-5, 
                    redux=1, fix_end_points=False, fix_rng=None, 
                    verbose=True)
    y_als = als.calculate(y)

    assert np.max((y - y_als)[200:800]) > 0.002

    als = AlsCvxopt(smoothness_param=1e3, asym_param=1e-5, 
                    redux=1, fix_end_points=False, fix_rng=fix_rng, 
                    verbose=True)
    y_als = als.calculate(y)
    assert np.max((y - y_als)[200:800]) < 0.002
示例#5
0
def test_rng_redux_fix_rng():
    x = np.linspace(-100, 100, 1000)
    y = 10*np.exp(-(x**2/(2*20**2)))

    als = AlsCvxopt(smoothness_param=1, asym_param=1e-7, 
                    redux=10, fix_end_points=False, fix_rng=None, 
                    verbose=True)
    y_als = als.calculate(y)

    assert np.max(y[250:750] - y_als[250:750]) > 9

    fix_rng = np.arange(200,800)
    
    als = AlsCvxopt(smoothness_param=1, asym_param=1e-7, 
                    redux=10, fix_end_points=False, fix_rng=fix_rng, 
                    verbose=True)
    y_als = als.calculate(y)
    assert np.max(y[250:750] - y_als[250:750]) < 0.004
示例#6
0
def test_basic_redux_rng():
    x = np.linspace(-100, 100, 1000)
    y = 10 * np.exp(-(x**2 / (2 * 20**2)))

    rng = np.arange(200, 800)
    als = AlsCvxopt(smoothness_param=1e-2,
                    asym_param=1e-1,
                    rng=rng,
                    redux=10,
                    fix_end_points=False,
                    fix_rng=None,
                    verbose=True)

    y_als = als.calculate(y)

    np.testing.assert_allclose(y_als[:200], 0)
    np.testing.assert_allclose(y_als[800:], 0)
    np.testing.assert_almost_equal(y[..., rng], y_als[..., rng], decimal=3)
示例#7
0
def test_fix_ends():
    x = np.linspace(-100, 100, 1000)
    y = 1000*np.exp(-x**2/(2*20**2)) + 200*np.exp(-(x+99)**2/(2*1**2))

    als = AlsCvxopt(smoothness_param=1e2, asym_param=1e-3, 
                    redux=1, fix_end_points=False, fix_rng=None, 
                    verbose=True)

    y_als = als.calculate(y)

    assert (y[0] - y_als[0] > 20)

    als = AlsCvxopt(smoothness_param=1e1, asym_param=1e-3, 
                    redux=1, fix_end_points=True, fix_rng=None, 
                    verbose=True)

    y_als = als.calculate(y)

    assert (y[0] - y_als[0] < 2)
示例#8
0
def test_vec_asym_param():
    x = np.linspace(-100, 100, 1000)
    y = 10*np.exp(-(x**2/(2*20**2)))

    asym_param = 1e-6*np.ones((x.size))

    als = AlsCvxopt(smoothness_param=1e3, asym_param=asym_param, 
                    redux=1, fix_end_points=False, fix_rng=None, 
                    verbose=True)
    y_als = als.calculate(y)

    assert np.max((y - y_als)[250:750]) > 7 

    asym_param[200:800] = 1e-2

    als = AlsCvxopt(smoothness_param=1e3, asym_param=asym_param, 
                    redux=1, fix_end_points=False, fix_rng=None, 
                    verbose=True)
    y_als = als.calculate(y)

    assert np.max((y - y_als)[250:750]) < 0.03
示例#9
0
def test_vec_asym_param():
    x = np.linspace(-100, 100, 1000)
    y = 10 * np.exp(-(x**2 / (2 * 20**2)))

    asym_param = 1e-6 * np.ones((x.size))

    als = AlsCvxopt(smoothness_param=1e3,
                    asym_param=asym_param,
                    redux=1,
                    fix_end_points=False,
                    fix_rng=None,
                    verbose=True)
    y_als = als.calculate(y)

    assert np.max((y - y_als)[250:750]) > 7

    asym_param[200:800] = 1e-2

    als = AlsCvxopt(smoothness_param=1e3,
                    asym_param=asym_param,
                    redux=1,
                    fix_end_points=False,
                    fix_rng=None,
                    verbose=True)
    y_als = als.calculate(y)

    assert np.max((y - y_als)[250:750]) < 0.03
示例#10
0
def test_fix_ends():
    x = np.linspace(-100, 100, 1000)
    y = 1000 * np.exp(-x**2 / (2 * 20**2)) + 200 * np.exp(-(x + 99)**2 /
                                                          (2 * 1**2))

    als = AlsCvxopt(smoothness_param=1e2,
                    asym_param=1e-3,
                    redux=1,
                    fix_end_points=False,
                    fix_rng=None,
                    verbose=True)

    y_als = als.calculate(y)

    assert (y[0] - y_als[0] > 20)

    als = AlsCvxopt(smoothness_param=1e1,
                    asym_param=1e-3,
                    redux=1,
                    fix_end_points=True,
                    fix_rng=None,
                    verbose=True)

    y_als = als.calculate(y)

    assert (y[0] - y_als[0] < 2)
示例#11
0
def test_rng_redux_fix_rng_vecasym():
    x = np.linspace(-100, 100, 1000)
    y = 10 * np.exp(-(x**2 / (2 * 20**2)))

    rng = np.arange(200, 800)

    asym_vec = 0 * x + 1e-7
    als = AlsCvxopt(smoothness_param=1,
                    asym_param=asym_vec,
                    rng=rng,
                    redux=10,
                    fix_end_points=False,
                    fix_rng=None,
                    verbose=True)
    y_als = als.calculate(y)

    assert np.max(y[250:750] - y_als[250:750]) > 9

    fix_rng = np.arange(600)
    # fix_rng = np.hstack((np.arange(100), np.arange(500,600)))

    als = AlsCvxopt(smoothness_param=1,
                    asym_param=1e-7,
                    rng=rng,
                    redux=10,
                    fix_end_points=False,
                    fix_rng=fix_rng,
                    verbose=True)
    y_als = als.calculate(y)
    assert np.max(y[250:750] - y_als[250:750]) < 0.004
示例#12
0
def test_fix_rng():
    x = np.linspace(-100, 100, 1000)
    y = 10 * np.exp(-(x**2 / (2 * 20**2)))
    fix_rng = np.arange(200, 800)

    als = AlsCvxopt(smoothness_param=1e3,
                    asym_param=1e-5,
                    redux=1,
                    fix_end_points=False,
                    fix_rng=None,
                    verbose=True)
    y_als = als.calculate(y)

    assert np.max((y - y_als)[200:800]) > 0.002

    als = AlsCvxopt(smoothness_param=1e3,
                    asym_param=1e-5,
                    redux=1,
                    fix_end_points=False,
                    fix_rng=fix_rng,
                    verbose=True)
    y_als = als.calculate(y)
    assert np.max((y - y_als)[200:800]) < 0.002