def test_collect_data(self):
start_time = '1/1/2015 13:00:00'
final_time = '1/2/2015'
# Instantiate control object
control = exodata.ControlFromDF(self.df, \
self.variable_map)
# Get control data
control.collect_data(start_time, final_time)
# Check reference
df_test = control.display_data()
self.check_df(df_test, 'collect_data.csv')
def setUp(self):
self.df = pd.read_csv(
os.path.join(self.get_unittest_path(), 'resources', 'building',
'ControlCSV_0.csv'))
time = pd.to_datetime(self.df['Time'])
self.df.set_index(time, inplace=True)
variable_map = {'conHeat_wes' : ('conHeat_wes', units.unit1), \
'conHeat_hal' : ('conHeat_hal', units.unit1), \
'conHeat_eas' : ('conHeat_eas', units.unit1)}
# Instantiate control object
self.control = exodata.ControlFromDF(self.df, \
variable_map)
def test_collect_data_tz_handling(self):
start_time = '1/1/2015 13:00:00'
final_time = '1/2/2015'
# Localize timezone
self.df = self.df.tz_localize('UTC')
# Instantiate weather object
with self.assertRaises(TypeError):
control = exodata.ControlFromDF(self.df, \
self.variable_map)
# Remove timezone
self.df = self.df.tz_convert(None)
# Instantiate weather object
control = exodata.ControlFromDF(self.df, \
self.variable_map)
# Get control data
control.collect_data(start_time, final_time)
# Collect twice
control.collect_data(start_time, final_time)
# Check reference
df_test = control.display_data()
self.check_df(df_test, 'collect_data.csv')
