Fixed retrieve hass test UTC

davidusb-geek · Jul 5, 2024 · bbb8d57 · bbb8d57
1 parent 3fee7a9
commit bbb8d57
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_retrieve_hass.py b/tests/test_retrieve_hass.py
@@ -122,7 +122,7 @@ def test_prepare_data(self):
         self.assertEqual(len(self.rh.df_final.columns), len(self.var_list))
         self.assertEqual(self.rh.df_final.index.isin(self.days_list).sum(), len(self.days_list))
         self.assertEqual(self.rh.df_final.index.freq, self.retrieve_hass_conf['freq'])
-        self.assertEqual(self.rh.df_final.index.tz, pytz.UTC)
+        self.assertEqual(self.rh.df_final.index.tz, datetime.timezone.utc)
         self.rh.prepare_data(self.retrieve_hass_conf['var_load'], 
                              load_negative = self.retrieve_hass_conf['load_negative'],
                              set_zero_min = self.retrieve_hass_conf['set_zero_min'],