def test_collect_data(self): start_time = '1/1/2015 13:00:00' final_time = '1/2/2015' # Instantiate weather object constraints = exodata.ConstraintFromDF(self.df, \ self.variable_map) # Get constraint data constraints.collect_data(start_time, final_time) # Check reference df_test = constraints.display_data() self.check_df(df_test, 'collect_data.csv')
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): constraints = exodata.ConstraintFromDF(self.df, \ self.variable_map) # Remove timezone self.df = self.df.tz_convert(None) # Instantiate weather object constraints = exodata.ConstraintFromDF(self.df, \ self.variable_map) # Get control data constraints.collect_data(start_time, final_time) # Collect twice constraints.collect_data(start_time, final_time) # Check reference df_test = constraints.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', 'optimization', 'sampleConstraintCSV_Setback.csv')) time = pd.to_datetime(self.df['Time']) self.df.set_index(time, inplace=True) variable_map = {'wesTdb_min' : ('wesTdb', 'GTE', units.degC), \ 'wesTdb_max' : ('wesTdb', 'LTE', units.degC), \ 'easTdb_min' : ('easTdb', 'GTE', units.degC), \ 'easTdb_max' : ('easTdb', 'LTE', units.degC), \ 'halTdb_min' : ('halTdb', 'GTE', units.degC), \ 'halTdb_max' : ('halTdb', 'LTE', units.degC), \ 'conHeat_wes_min' : ('conHeat_wes', 'GTE', units.unit1), \ 'conHeat_wes_max' : ('conHeat_wes', 'LTE', units.unit1), \ 'conHeat_hal_min' : ('conHeat_hal', 'GTE', units.unit1), \ 'conHeat_hal_max' : ('conHeat_hal', 'LTE', units.unit1), \ 'conHeat_eas_min' : ('conHeat_eas', 'GTE', units.unit1), \ 'conHeat_eas_max' : ('conHeat_eas', 'LTE', units.unit1)} # Instantiate weather object self.constraints = exodata.ConstraintFromDF(self.df, \ variable_map)