def are_answers_correctly_set(trial_conf, trial_data):
df_force_answer = np.array(trial_data[list(FORCE_CLASS_COLS)].iloc[0])
df_string_force_answer = FORCE_CLASS_COLS[df_force_answer.argmax()]
conf_string_force_answer = gps.get_force_answer(trial_conf[0]["lf"])
df_mass_answer = np.array(trial_data[list(MASS_CLASS_COLS)].iloc[0])
df_string_mass_answer = MASS_CLASS_COLS[df_mass_answer.argmax()]
conf_string_mass_answer = gps.get_mass_answer(trial_conf[0]["mass"])
return (conf_string_force_answer
== df_string_force_answer) and (conf_string_mass_answer
== df_string_mass_answer)
def test_empty_list(self):
masses = []
with self.assertRaises(IndexError):
gps.get_mass_answer(masses)
def test_list_too_short(self):
masses = [2]
with self.assertRaises(IndexError):
gps.get_mass_answer(masses)
def test_same(self):
masses = [5, 5]
self.assertEqual(gps.get_mass_answer(masses), 'same')
def test_b_is_larger(self):
masses = [6, 7, 5]
self.assertEqual(gps.get_mass_answer(masses), 'B')
def test_a_is_larger(self):
masses = [10, 8, 12, 90]
self.assertEqual(gps.get_mass_answer(masses), 'A')