from utils import utils, sus
from utils import data as data_module

%matplotlib inline

usability_data = sus.load_table()
usability_data = sus.fillna(usability_data)
a, b = data_module.to_groups(usability_data)
sus.per_question_plot(a, b)

data_module.save_to_graphics('usability-per_question_statistics')

scores = usability_data.apply(sus.score, axis=1)
a, b = data_module.to_groups(scores)
utils.independent_one_sided_ttest_summary(a, b, ylabel='SUS score', ylim=(0, 100))

data_module.save_to_graphics('usability-sus_scores')

means = a.mean(), b.mean()
studies_dataset = data_module.sus_dataset.multisurvey_studies
for group, mean in zip(['A', 'B'], means):
    dataset_std = (mean - studies_dataset['mean']) / studies_dataset['std']
    print('Group {} std from dataset: {}'.format(group, dataset_std))