Example #1
0
def test_step_wrong_not_scalar_offset():
    t_init = 0
    T = 5
    A = 1.5
    time = np.linspace(0, 10, 11)
    var = np.ones_like(time)
    offset = var

    with pytest.raises(TypeError) as excinfo:
        step(t_init, T, A, time, offset=offset)
def test_step_wrong_not_scalar_offset():
    t_init = 0
    T = 5
    A = 1.5
    time = np.linspace(0, 10, 11)
    var = np.ones_like(time)
    offset = var

    with pytest.raises(TypeError) as excinfo:
        step(t_init, T, A, time, offset=offset)
def test_step_wrong_size_var():
    t_init = 0
    T = 5
    A = 1.5
    time = np.linspace(0, 10, 11)
    var = np.ones([10])

    with pytest.raises(ValueError) as excinfo:
        step(t_init, T, A, time, offset=0, var=var)
    assert ("ValueError: var and time must have the same size"
            in excinfo.exconly())
Example #4
0
def test_step_wrong_size_var():
    t_init = 0
    T = 5
    A = 1.5
    time = np.linspace(0, 10, 11)
    var = np.ones([10])

    with pytest.raises(ValueError) as excinfo:
        step(t_init, T, A, time, offset=0, var=var)
    assert ("ValueError: var and time must have the same size"
                in excinfo.exconly())
def test_step_bounds_not_included():
    t_init = 0.1
    T = 4.8
    A = 1.5
    time = np.linspace(0, 10, 11)

    expected_input = np.zeros([11])
    expected_input[1:5] = A

    step_input = step(t_init, T, A, time, offset=0, var=None)

    assert_almost_equal(step_input, expected_input)
def test_step():
    t_init = 0
    T = 5
    A = 1.5
    time = np.linspace(0, 10, 11)

    expected_input = np.zeros([11])
    expected_input[0:6] = A

    step_input = step(t_init, T, A, time, offset=0, var=None)

    assert_almost_equal(step_input, expected_input)
Example #7
0
def test_step_bounds_not_included():
    t_init = 0.1
    T = 4.8
    A = 1.5
    time = np.linspace(0, 10, 11)

    expected_input = np.zeros([11])
    expected_input[1:5] = A

    step_input = step(t_init, T, A, time, offset=0, var=None)

    assert_almost_equal(step_input, expected_input)
Example #8
0
def test_step():
    t_init = 0
    T = 5
    A = 1.5
    time = np.linspace(0, 10, 11)

    expected_input = np.zeros([11])
    expected_input[0:6] = A

    step_input = step(t_init, T, A, time, offset=0, var=None)

    assert_almost_equal(step_input, expected_input)
def test_step_var():
    t_init = 0
    T = 5
    A = 1.5
    time = np.linspace(0, 10, 11)
    var = np.ones_like(time)
    var[0::2] = -1

    expected_input = var.copy()
    expected_input[0:6] += A

    step_input = step(t_init, T, A, time, offset=0, var=var)

    assert_almost_equal(step_input, expected_input)
Example #10
0
def test_step_var():
    t_init = 0
    T = 5
    A = 1.5
    time = np.linspace(0, 10, 11)
    var = np.ones_like(time)
    var[0::2] = -1

    expected_input = var.copy()
    expected_input[0:6] += A

    step_input = step(t_init, T, A, time, offset=0, var=var)

    assert_almost_equal(step_input, expected_input)
Example #11
0
def test_step_var_and_offset():
    t_init = 0
    T = 5
    A = 1.5
    time = np.linspace(0, 10, 11)
    offset = 2.6
    var = np.ones_like(time)
    var[0::2] = -1

    expected_input = var + offset
    expected_input[0:6] += A

    step_input = step(t_init, T, A, time, offset=offset, var=var)

    assert_almost_equal(step_input, expected_input)
Example #12
0
def test_step_var_and_offset():
    t_init = 0
    T = 5
    A = 1.5
    time = np.linspace(0, 10, 11)
    offset = 2.6
    var = np.ones_like(time)
    var[0::2] = -1

    expected_input = var + offset
    expected_input[0:6] += A

    step_input = step(t_init, T, A, time, offset=offset, var=var)

    assert_almost_equal(step_input, expected_input)