def test_selected_input_array(self):
action_selection = DDM(
input_format=ARRAY,
function=DriftDiffusionAnalytical(
),
output_ports=[SELECTED_INPUT_ARRAY],
name='DDM'
)
action_selection.execute([1.0])
def test_DDM_noise_2_0():
stim = 10
T = DDM(name='DDM',
function=DriftDiffusionIntegrator(noise=2.0,
rate=1.0,
time_step_size=1.0))
val = float(T.execute(stim)[0])
assert val == 11.34362792551828
def test_DDM_time():
D = DDM(name='DDM',
function=DriftDiffusionIntegrator(noise=0.0,
rate=-5.0,
time_step_size=0.2,
starting_point=0.5))
time_0 = D.function.previous_time # t_0 = 0.5
np.testing.assert_allclose(time_0, 0.5, atol=1e-08)
time_1 = D.execute(10)[1][0] # t_1 = 0.5 + 0.2 = 0.7
np.testing.assert_allclose(time_1[0], 0.7, atol=1e-08)
for i in range(10): # t_11 = 0.7 + 10*0.2 = 2.7
D.execute(10)
time_12 = D.execute(10)[1][0] # t_12 = 2.7 + 0.2 = 2.9
np.testing.assert_allclose(time_12[0], 2.9, atol=1e-08)
def test_DDM_input(stim):
T = DDM(
name='DDM',
function=DriftDiffusionIntegrator(noise=0.0,
rate=1.0,
time_step_size=1.0),
)
val = float(T.execute(stim)[0])
assert val == 10
def test_DDM_rate_list_len_1():
stim = 10
T = DDM(
name='DDM',
function=DriftDiffusionIntegrator(noise=0.0,
rate=[5],
time_step_size=1.0),
)
val = float(T.execute(stim)[0])
assert val == 50
def test_DDM_noise_0_5():
stim = 10
T = DDM(name='DDM',
function=DriftDiffusionIntegrator(noise=0.5,
rate=1.0,
time_step_size=1.0))
val = float(T.execute(stim)[0])
assert val == 10.67181396275914
def test_DDM_input_fn():
with pytest.raises(TypeError) as error_text:
stim = NormalDist()
T = DDM(
name='DDM',
function=DriftDiffusionIntegrator(noise=0.0,
rate=1.0,
time_step_size=1.0),
)
float(T.execute(stim))
assert "not supported for the input types" in str(error_text.value)
def test_DDM_input_rate_negative():
stim = [10]
T = DDM(
name='DDM',
default_variable=[0],
function=DriftDiffusionIntegrator(noise=0.0,
rate=-5.0,
time_step_size=1.0),
)
val = float(T.execute(stim)[0])
assert val == -50
def test_DDM_rate_fn():
with pytest.raises(typecheck.framework.InputParameterError) as error_text:
stim = [10]
T = DDM(
name='DDM',
default_variable=[0],
function=DriftDiffusionIntegrator(noise=0.0,
rate=NormalDist().function,
time_step_size=1.0),
)
float(T.execute(stim)[0])
assert "incompatible value" in str(error_text.value)
def test_DDM_input_list_len_2():
with pytest.raises(DDMError) as error_text:
stim = [10, 10]
T = DDM(
name='DDM',
default_variable=[0, 0],
function=DriftDiffusionIntegrator(noise=0.0,
rate=1.0,
time_step_size=1.0),
)
float(T.execute(stim)[0])
assert "single numeric item" in str(error_text.value)
def test_DDM_noise_invalid(noise):
with pytest.raises(FunctionError) as error_text:
stim = 10
T = DDM(
name='DDM',
function=DriftDiffusionIntegrator(noise=noise,
rate=1.0,
time_step_size=1.0),
)
float(T.execute(stim)[0])
assert "DriftDiffusionIntegrator requires noise parameter to be a float" in str(
error_text.value)
def test_DDM_size_int_inputs():
T = DDM(
name='DDM',
size=1,
function=DriftDiffusionIntegrator(noise=0.0,
rate=-5.0,
time_step_size=1.0),
)
val = T.execute([.4])
decision_variable = val[0][0]
time = val[1][0]
assert decision_variable == -2.0
assert time == 1.0
def test_threshold_stops_accumulation_negative(self):
D = DDM(name='DDM', function=DriftDiffusionIntegrator(threshold=5.0))
decision_variables = []
time_points = []
for i in range(5):
output = D.execute(-2.0)
decision_variables.append(output[0][0][0])
time_points.append(output[1][0][0])
# decision variable accumulation stops
assert np.allclose(decision_variables, [-2.0, -4.0, -5.0, -5.0, -5.0])
# time accumulation does not stop
assert np.allclose(time_points, [1.0, 2.0, 3.0, 4.0, 5.0])
def test_threshold_sets_is_finished(self):
D = DDM(name='DDM', function=DriftDiffusionIntegrator(threshold=5.0))
D.execute(2.0) # 2.0 < 5.0
assert not D.is_finished()
D.execute(2.0) # 4.0 < 5.0
assert not D.is_finished()
D.execute(2.0) # 5.0 = threshold
assert D.is_finished()
def test_valid(self):
D = DDM(
name='DDM',
function=DriftDiffusionIntegrator(seed=0),
)
# returns previous_value + rate * variable * time_step_size + noise
# 0.0 + 1.0 * 1.0 * 1.0 + 0.0
D.execute(1.0)
assert np.allclose(np.asfarray(D.value), [[1.0], [1.0]])
assert np.allclose(D.output_ports[0].value[0][0], 1.0)
assert np.allclose(D.output_ports[1].value[0][0], 1.0)
# reinitialize function
D.function.reinitialize(2.0, 0.1, 0)
assert np.allclose(D.function.value[0], 2.0)
assert np.allclose(D.function.previous_value, 2.0)
assert np.allclose(D.function.previous_time, 0.1)
assert np.allclose(np.asfarray(D.value), [[1.0], [1.0]])
assert np.allclose(D.output_ports[0].value[0][0], 1.0)
assert np.allclose(D.output_ports[1].value[0][0], 1.0)
# reinitialize function without value spec
D.function.reinitialize()
assert np.allclose(D.function.value[0], 0.0)
assert np.allclose(D.function.previous_value, 0.0)
assert np.allclose(D.function.previous_time, 0.0)
assert np.allclose(np.asfarray(D.value), [[1.0], [1.0]])
assert np.allclose(D.output_ports[0].value[0][0], 1.0)
assert np.allclose(D.output_ports[1].value[0][0], 1.0)
# reinitialize mechanism
D.reinitialize(2.0, 0.1, 0)
assert np.allclose(D.function.value[0], 2.0)
assert np.allclose(D.function.previous_value, 2.0)
assert np.allclose(D.function.previous_time, 0.1)
assert np.allclose(np.asfarray(D.value), [[2.0], [0.1]])
assert np.allclose(D.output_ports[0].value, 2.0)
assert np.allclose(D.output_ports[1].value, 0.1)
D.execute(1.0)
# 2.0 + 1.0 = 3.0 ; 0.1 + 1.0 = 1.1
assert np.allclose(np.asfarray(D.value), [[[3.0]], [[1.1]]])
assert np.allclose(D.output_ports[0].value[0][0], 3.0)
assert np.allclose(D.output_ports[1].value[0][0], 1.1)
# reinitialize mechanism without value spec
D.reinitialize()
assert np.allclose(D.function.value[0], 0.0)
assert np.allclose(D.function.previous_value, 0.0)
assert np.allclose(D.function.previous_time, 0.0)
assert np.allclose(D.output_ports[0].value[0], 0.0)
assert np.allclose(D.output_ports[1].value[0], 0.0)
# reinitialize only decision variable
D.function.initializer = 1.0
D.function.starting_point = 0.0
D.reinitialize()
assert np.allclose(D.function.value[0], 1.0)
assert np.allclose(D.function.previous_value, 1.0)
assert np.allclose(D.function.previous_time, 0.0)
assert np.allclose(D.output_ports[0].value[0], 1.0)
assert np.allclose(D.output_ports[1].value[0], 0.0)
def test_DDM_Integrator_Bogacz():
stim = 10
T = DDM(name='DDM', function=DriftDiffusionAnalytical())
val = float(T.execute(stim)[0])
assert val == 1.0
