def test___calculate_go_errors():
pid = "104"
df = get_sart_trial_block(pid)
# check known values
assert list(df.correct.values).count(False) == 5
df = compile_data._add_anticipation_errors(df)
# check known values
assert list(df.anticipate_error.values).count(True) == 6
assert list(df.correct.values).count(False) == 11
go_errors = compile_data._calculate_go_errors(df, 'go')
assert isinstance(go_errors, compile_data.pd.Series)
assert list(go_errors).count(True) == 1
nogo_errors = compile_data._calculate_go_errors(df, 'no_go')
assert isinstance(nogo_errors, compile_data.pd.Series)
assert list(nogo_errors).count(True) == 4
def test__calculate_nogo_error_rt_avgs_011():
pid = "011"
df = get_sart_trial_block(pid)
df = compile_data._add_anticipation_errors(df)
df['nogo_error'] = compile_data._calculate_go_errors(df, 'no_go')
assert list(df['nogo_error']).count(True) == 4
adjacent_rts = compile_data._calculate_nogo_error_rt_avgs(df)
assert adjacent_rts['prev4_avg'] == 381.444444444
assert adjacent_rts['num_prev4_rts'] == 9
assert adjacent_rts['next4_avg'] == 391.875
assert adjacent_rts['num_next4_rts'] == 8
def test__calculate_nogo_error_rt_avgs_104():
pid = "104"
df = get_sart_trial_block(pid)
df = compile_data._add_anticipation_errors(df)
df['nogo_error'] = compile_data._calculate_go_errors(df, 'no_go')
assert list(df['nogo_error']).count(True) == 4
adjacent_rts = compile_data._calculate_nogo_error_rt_avgs(df)
assert adjacent_rts['prev4_avg'] == 254.4375
assert adjacent_rts['num_prev4_rts'] == 16
assert adjacent_rts['next4_avg'] == 224.1875
assert adjacent_rts['num_next4_rts'] == 16
