fix: test all curtailment columns

Breakthrough-Energy · Jun 1, 2020 · f34573f · f34573f
1 parent fdde9ed
commit f34573f
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/postreise/analyze/generation/tests/test_curtailment.py b/postreise/analyze/generation/tests/test_curtailment.py
@@ -60,12 +60,12 @@ def _check_curtailment_vs_expected(self, curtailment, expected):
         self.assertEqual(curtailment.keys(), expected.keys())
         for key in curtailment.keys():
             self.assertIsInstance(curtailment[key], pd.DataFrame)
-        assert_array_equal(curtailment[key].index.to_numpy(),
-                           expected[key].index.to_numpy())
-        assert_array_equal(curtailment[key].columns.to_numpy(),
-                           expected[key].columns.to_numpy())
-        assert_array_equal(curtailment[key].to_numpy(),
-                           expected[key].to_numpy())
+            assert_array_equal(curtailment[key].index.to_numpy(),
+                               expected[key].index.to_numpy())
+            assert_array_equal(curtailment[key].columns.to_numpy(),
+                               expected[key].columns.to_numpy())
+            assert_array_equal(curtailment[key].to_numpy(),
+                               expected[key].to_numpy())
 
     def test_calculate_curtailment_time_series_solar(self):
         expected_return = {'solar': mock_curtailment['solar']}