def test_questionnaire_grouped_summary():
df = setup_dataframe.create_categorical_dataframe()
fig = EDA_categorical.questionnaire_grouped_summary(df, question='id_1', id_column ='id',\
answer_column='answer',group = 'group',title=None,\
xlabel=None,)
assert (type(fig) == plotly.graph_objs._figure.Figure)
def test_plot_grouped_categories():
df = setup_dataframe.create_categorical_dataframe()
grouped = EDA_categorical.question_by_group(df, question = 'id_1', id_column = 'id',\
answer_column = 'answer', group='group')
fig = EDA_categorical.plot_grouped_categories(grouped, group='group', title=None, xlabel=None, \
ylabel=None, width=900, height=900)
assert (type(fig) == plotly.graph_objs._figure.Figure)
def test_question_by_group():
df = setup_dataframe.create_categorical_dataframe()
grouped = EDA_categorical.question_by_group(df, question = 'id_1', id_column = 'id',\
answer_column = 'answer', group='group')
#TODO! test contents of groupby object
assert (type(grouped) == pd.core.frame.DataFrame)
def test_questionnaire_summary_one_subject():
df = setup_dataframe.create_categorical_dataframe()
fig = EDA_categorical.questionnaire_summary(df, 'id_1', 'answer', title=None,\
xlabel=None, ylabel=None, user='******',\
width=900, height=900)
assert (type(fig) == plotly.graph_objs._figure.Figure)
def test_plot_categories():
df = setup_dataframe.create_categorical_dataframe()
answers = EDA_categorical.categorize_answers(df, 'id', 'answer')
fig = EDA_categorical.plot_categories(answers, title=None, xlabel=None, \
ylabel=None, width=900, height=900)
assert (type(fig) == plotly.graph_objs._figure.Figure)
def test_categorize_answers():
df = setup_dataframe.create_categorical_dataframe()
answers = EDA_categorical.categorize_answers(df, 'id', 'answer')
#TODO! fix this
#assert (answers.index.tolist() == ['str_1'])
assert np.allclose(answers.values, np.array([3]))