def update_model_state(add_displacement_timestamp, add_regression_timestamp,
remove_model_timestamp, displacement_value,
displacement_unit, plotted_models,
remove_model_selection, regression_model_choice,
training_start_date, training_end_date,
regression_model_name):
return model_state.update(load_data, regression_models,
add_displacement_timestamp,
add_regression_timestamp, remove_model_timestamp,
displacement_value, displacement_unit,
plotted_models, remove_model_selection,
regression_model_choice, training_start_date,
training_end_date, regression_model_name)
def test_initial_empty_model_list(self):
add_displacement_timestamp = '0'
add_regression_timestamp = '0'
remove_model_timestamp = '0'
displacement_value = 0
displacement_unit = 'halfhours'
remove_model_selection = ''
plotted_models = []
training_start_date = 'None'
training_end_date = 'None'
regression_model_choice = 'None'
regression_model_name = ''
expected_state = json.dumps([])
actual_state = update(self.load_data, self.regression_models, add_displacement_timestamp, add_regression_timestamp, remove_model_timestamp, displacement_value,
displacement_unit, plotted_models, remove_model_selection, regression_model_choice, training_start_date,
training_end_date, regression_model_name)
self.assertEqual(actual_state, expected_state)
def test_initial_empty_model_list_add_new_displacement_model(self):
add_displacement_timestamp = '1'
add_regression_timestamp = '0'
remove_model_timestamp = '0'
displacement_value = 5
displacement_unit = 'halfhours'
plotted_models = json.dumps([])
remove_model_selection = ''
training_start_date = 'None'
training_end_date = 'None'
regression_model_choice = 'None'
regression_model_name = ''
expected_state = json.dumps([dict(model_name='-5 halfhours', model_type='displacement', model_color='color1', displacement_value=5,
displacement_unit='halfhours')])
actual_state = update(self.load_data, self.regression_models, add_displacement_timestamp, add_regression_timestamp, remove_model_timestamp, displacement_value,
displacement_unit, plotted_models, remove_model_selection, regression_model_choice, training_start_date,
training_end_date, regression_model_name)
self.assertEqual(actual_state, expected_state)
def test_existing_models_list_add_new_regression_model(self, mock_function):
mock_function.return_value = dict(coefficients=[111.0], intercept=0)
add_displacement_timestamp = '0'
add_regression_timestamp = '1'
remove_model_timestamp = '0'
displacement_value = 10
displacement_unit = 'hours'
plotted_models = json.dumps([dict(model_name='-5 halfhours', model_type='displacement', model_color='color1', displacement_value=5,
displacement_unit='halfhours')])
remove_model_selection = ''
regression_model_choice = 'model1'
training_start_date = '2018-01-02'
training_end_date = '2018-01-03'
regression_model_name = 'custom_model_name'
expected_state = json.dumps([dict(model_name='-5 halfhours', model_type='displacement', model_color='color1', displacement_value=5,
displacement_unit='halfhours'), dict(model_name='custom_model_name', model_type='regression', model_color='color1', coefficients=[111.0], intercept=0,
x_columns=["Temperature"], corrected_column=None)])
actual_state = update(self.load_data, self.regression_models, add_displacement_timestamp, add_regression_timestamp, remove_model_timestamp, displacement_value,
displacement_unit, plotted_models, remove_model_selection, regression_model_choice, training_start_date,
training_end_date, regression_model_name)
self.assertEqual(actual_state, expected_state)
def test_existing_models_list_remove_model(self):
add_displacement_timestamp = '0'
add_regression_timestamp = '0'
remove_model_timestamp = '1'
displacement_value = 0
displacement_unit = 'hours'
plotted_models = json.dumps([dict(model_name='-5 halfhours', model_type='displacement', model_color='color1', displacement_value=5,
displacement_unit='halfhours'),
dict(model_name='model1', model_type='regression', model_color='color2',
coefficients=[111.0], intercept=0, corrected_column=None)])
remove_model_selection = '-5 halfhours'
regression_model_choice = 'None'
training_start_date = 'None'
training_end_date = 'None'
regression_model_name = 'model1'
expected_state = json.dumps([dict(model_name='model1', model_type='regression', model_color='color2',
coefficients=[111.0], intercept=0, corrected_column=None)])
actual_state = update(self.load_data, self.regression_models, add_displacement_timestamp, add_regression_timestamp, remove_model_timestamp, displacement_value,
displacement_unit, plotted_models, remove_model_selection, regression_model_choice, training_start_date,
training_end_date, regression_model_name)
self.assertEqual(actual_state, expected_state)