def test_collect_data_tz_handling(self): start_time = '2016-10-19 19:53:00' final_time = '2016-10-20 06:53:00' time = pd.to_datetime(self.df['DateUTC']) self.df.set_index(time, inplace=True) # Localize timezone self.df = self.df.tz_localize('UTC') # Instantiate weather object with self.assertRaises(TypeError): weather = exodata.WeatherFromDF(self.df, \ self.variable_map, \ self.geography) # Remove timezone self.df = self.df.tz_convert(None) # Instantiate weather object weather = exodata.WeatherFromDF(self.df, \ self.variable_map, \ self.geography) # Get weather data weather.collect_data(start_time, final_time) # Collect twice weather.collect_data(start_time, final_time) # Check reference df_test = weather.display_data() self.check_df(df_test, 'collect_data_default_time.csv')
def test_instantiate(self): time = pd.to_datetime(self.df['DateUTC']) self.df.set_index(time, inplace=True) weather = exodata.WeatherFromDF(self.df, \ self.variable_map, \ geography = self.geography) self.assertEqual(weather.name, 'weather_from_df') self.assertAlmostEqual(weather.lat.display_data(), 37.8716, places=4) self.assertAlmostEqual(weather.lon.display_data(), -122.2727, places=4) self.assertEqual(weather.tz_name, 'UTC')
def test_collect_data_default_time(self): start_time = '2016-10-19 19:53:00' final_time = '2016-10-20 06:53:00' time = pd.to_datetime(self.df['DateUTC']) self.df.set_index(time, inplace=True) # Instantiate weather object weather = exodata.WeatherFromDF(self.df, \ self.variable_map, \ geography = self.geography) # Get weather data weather.collect_data(start_time, final_time) # Check reference df_test = weather.display_data() self.check_df(df_test, 'collect_data_default_time.csv')
def test_collect_data_local_time_from_tz_name(self): start_time = '10/19/2016 12:53:00 PM' final_time = '10/19/2016 11:53:00 PM' time = pd.to_datetime(self.df['TimePDT']) self.df.set_index(time, inplace=True) # Instantiate weather object weather = exodata.WeatherFromDF(self.df, \ self.variable_map, \ geography = self.geography, tz_name = 'America/Los_Angeles') # Get weather data weather.collect_data(start_time, final_time) # Check reference df_test = weather.display_data() self.check_df(df_test, 'collect_data_local_time_from_tz_name.csv')
def test_collect_data_local_time_from_geography(self): start_time = '10/19/2016 12:53:00 PM'; final_time = '10/19/2016 11:53:00 PM'; time = pd.to_datetime(self.df['TimePDT']); self.df.set_index(time, inplace=True); # Instantiate weather object weather = exodata.WeatherFromDF(self.df, \ self.variable_map, \ self.geography, tz_name = 'from_geography'); # Get weather data weather.collect_data(start_time, final_time); # Check reference df_test = weather.display_data(); self.check_df(df_test, 'collect_data_local_time_from_geography.csv');
def test_instantiate_without_geography(self): with self.assertRaises(TypeError): weather = exodata.WeatherFromDF(self.df, self.variable_map)