refactor: assign timestamp to PG data frame in test

Breakthrough-Energy · Sep 12, 2020 · f2477f0 · f2477f0
1 parent f16fcdb
commit f2477f0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/postreise/analyze/generation/tests/test_carbon.py b/postreise/analyze/generation/tests/test_carbon.py
@@ -153,7 +153,8 @@ def setUp(self):
             {
                 plant_id: [(i + 1) * p for p in range(self.period_num)]
                 for i, plant_id in enumerate(mock_plant["plant_id"])
-            }
+            },
+            index=pd.date_range(start="2016-01-01", periods=self.period_num, freq="H"),
         )
         self.plant = pd.DataFrame(mock_plant)
         self.plant.set_index("plant_id", inplace=True)