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())
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)
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)
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)