From 7aeb40d674f225abc8385e34c9b83d5a42839ad0 Mon Sep 17 00:00:00 2001
From: junos <junos.lukan@ijs.si>
Date: Thu, 22 Jul 2021 15:47:42 +0200
Subject: [PATCH] Fix formatting.

---
 features/screen.py |  2 +-
 test/test_esm.py   | 44 +++++++++++++++++++++++++++++++++++++-------
 2 files changed, 38 insertions(+), 8 deletions(-)

diff --git a/features/screen.py b/features/screen.py
index e2e3d2c..593fadc 100644
--- a/features/screen.py
+++ b/features/screen.py
@@ -49,4 +49,4 @@ def time_screen_sequence(df_screen: pd.DataFrame) -> pd.DataFrame:
     # For example, from the two main sequences outlined above, the time of "real" phone usage can be calculated,
     #   i.e. how long the screen was unlocked.
     # Another example might be the average time between screen unlocks and/or screen status checks.
-    pass
\ No newline at end of file
+    pass
diff --git a/test/test_esm.py b/test/test_esm.py
index 84e573c..1bf5fb9 100644
--- a/test/test_esm.py
+++ b/test/test_esm.py
@@ -19,14 +19,24 @@ class EsmFeatures(unittest.TestCase):
         self.assertIn("question_id", self.esm_processed)
 
     def test_classify_sessions_by_completion(self):
-        self.esm_classified_sessions = classify_sessions_by_completion(self.esm_processed)
+        self.esm_classified_sessions = classify_sessions_by_completion(
+            self.esm_processed
+        )
         self.assertFalse(self.esm_classified_sessions["session_response"].isna().any())
         # Test that all sessions were indeed classified.
 
     def test_classify_sessions_by_completion_time(self):
         self.esm_classified = classify_sessions_by_completion_time(self.esm_processed)
-        session_of_interest = self.esm_processed.query("(device_id == '049df3f8-8541-4cf5-af2b-83f6b3f0cf4b') & (esm_session == 1)").sort_values("_id").reset_index()
-        session_of_interest_reclassified = self.esm_classified.query("(device_id == '049df3f8-8541-4cf5-af2b-83f6b3f0cf4b') & (esm_session == 1)").reset_index()
+        session_of_interest = (
+            self.esm_processed.query(
+                "(device_id == '049df3f8-8541-4cf5-af2b-83f6b3f0cf4b') & (esm_session == 1)"
+            )
+            .sort_values("_id")
+            .reset_index()
+        )
+        session_of_interest_reclassified = self.esm_classified.query(
+            "(device_id == '049df3f8-8541-4cf5-af2b-83f6b3f0cf4b') & (esm_session == 1)"
+        ).reset_index()
         self.assertEqual(session_of_interest.loc[0, "time"], "morning")
         self.assertEqual(session_of_interest_reclassified.loc[0, "time"], "daytime")
         # Check that the first (morning) session is reclassified as a daytime session.
@@ -38,10 +48,30 @@ class EsmFeatures(unittest.TestCase):
     def test_reverse_jcq_demand_control_scoring(self):
         esm_reversed = reverse_jcq_demand_control_scoring(self.esm_clean)
 
-        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"]))
+        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"]))
+        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.