def test_failing_compile_practice_data():
practice_path_7 = os.path.join(MOCK_DATA_DIR, 'practice', '7.csv')
df = compile_data.get_csv_as_dataframe(practice_path_7)
data = compile_data.compile_practice_data(df)
assert data['id'] == 7
assert data['passed_practice'] == False
assert data['time_practice_ms'] == 334489
def get_csv_as_df(stage, pid):
"""Take an experiment stage and participant ID and return a pandas
data frame.
"""
csv_path = os.path.join(MOCK_DATA_DIR, stage, '{}.csv'.format(pid))
df = compile_data.get_csv_as_dataframe(csv_path)
return df
def test_passing_compile_practice_data():
df = compile_data.get_csv_as_dataframe(PRACTICE_PATH_1)
data = compile_data.compile_practice_data(df)
assert data['id'] == 1
assert data['passed_practice'] == True
assert data['time_practice_blk1_ms'] == 33045
assert data['time_practice_blk2_ms'] == 83144
assert data['time_practice_ms'] == 151671
def test_extract_sart_blocks_with_2_practice():
# NOTE: tests out get_csv_as_dataframe() from compile_data
csv_path = _csv_path('practice', '003')
df = compile_data.get_csv_as_dataframe(csv_path)
blocks = compile_data.extract_sart_blocks(df)
assert len(blocks) == 2
for b in blocks:
assert isinstance(b, compile_data.pd.DataFrame)
# number of trials
assert len(blocks[0].index.values) == 5
assert len(blocks[1].index.values) == 15