diff --git a/test/test_esm.py b/test/test_esm.py index 6ee7513..84e573c 100644 --- a/test/test_esm.py +++ b/test/test_esm.py @@ -37,6 +37,11 @@ class EsmFeatures(unittest.TestCase): def test_reverse_jcq_demand_control_scoring(self): esm_reversed = reverse_jcq_demand_control_scoring(self.esm_clean) - print(esm_reversed.loc[esm_reversed["question_id"] == 73, "esm_user_score"]) - #self.assertIn(1.0, esm_reversed.loc[esm_reversed["question_id"] == 73, "esm_user_score"]) - #self.assertNotIn(0, esm_reversed.loc[esm_reversed["question_id"] == 73, "esm_user_score"]) \ No newline at end of file + + self.assertEqual(0, int(self.esm_clean.loc[self.esm_clean["question_id"] == 73, "esm_user_answer_numeric"])) + self.assertEqual(1, int(esm_reversed.loc[esm_reversed["question_id"] == 73, "esm_user_score"])) + # An example of a regular item: the score gets incremented by 1, to shift to 1-4 scoring. + + self.assertEqual(0, int(self.esm_clean.loc[self.esm_clean["question_id"] == 79, "esm_user_answer_numeric"])) + self.assertEqual(4, int(esm_reversed.loc[esm_reversed["question_id"] == 79, "esm_user_score"])) + # An example of a reversed item.