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)
