formatting

Author: nmdefries

google_symptoms/tests/test_date_utils.py

@@ -86,6 +86,10 @@ def generate_expected_start_end_dates(self, params_, issue_date):
 
         dates = dates_dict[issue_date]
 
+        # Raw signals add 6 extra dates of padding for later calculating
+        # smoothed signals. Since this test is checking an early step in the
+        # process, before padding has happened, we can drop the first 6
+        # dates.
         return {
             "export_start_date": min(dates[6:21]),
             "export_end_date": max(dates[6:21])

google_symptoms/tests/test_patch.py

@@ -46,6 +46,7 @@ def generate_expected_dates(self, params_, smoother, issue_date):
         if smoother == "raw":
             return dates
         else:
+            # Smoothed signals drop the first 6 dates.
             return dates[6:21]
 
     def mocked_patch(self, params_):
@@ -79,11 +80,15 @@ def side_effect(*args, **kwargs):
 
                 assert smoothed_dates == expected_smoothed_dates
                 assert raw_dates == expected_raw_dates
+
                 shutil.rmtree(issue_dir)
+
                 start_date += timedelta(days=1)
+
     def test_patch_default(self, params_w_patch):
         params_w_patch["indicator"]["num_export_days"] = None
         self.mocked_patch(params_w_patch)
+
     def test_patch_date_set(self, params_w_patch):
         self.mocked_patch(params_w_patch)