def test_calc_selected_dont_overwrite_dont_skip_nulls( self, mock_selected_obsids, mock_messagebar, mock_skippopup): mock_selected_obsids.return_value = ['rb1', 'rb2'] db_utils.sql_alter_db( '''INSERT INTO obs_points (obsid, h_toc) VALUES ('rb1', 1)''') db_utils.sql_alter_db( '''INSERT into w_levels (obsid, meas, date_time) VALUES ('rb1', 222, '2005-01-01 00:00:00')''' ) db_utils.sql_alter_db( '''INSERT INTO obs_points (obsid, h_toc) VALUES ('rb2', NULL)''') db_utils.sql_alter_db( '''INSERT into w_levels (obsid, meas, date_time) VALUES ('rb2', 444, '2005-01-01 00:00:00')''' ) db_utils.sql_alter_db( '''INSERT into w_levels (obsid, meas, level_masl, date_time) VALUES ('rb2', 555, 667, '2005-01-02 00:00:00')''' ) self.calclvl.FromDateTime = QtWidgets.QDateTimeEdit() self.calclvl.FromDateTime.setDateTime( datestring_to_date('2000-01-01 00:00:00')) self.calclvl.ToDateTime = QtWidgets.QDateTimeEdit() self.calclvl.ToDateTime.setDateTime( datestring_to_date('2010-01-01 00:00:00')) self.calclvl.checkBox_overwrite_prev.setChecked(False) self.calclvl.calcselected() #self.checkBox_skipnulls test_string = utils_for_tests.create_test_string( db_utils.sql_load_fr_db( 'SELECT obsid, date_time, meas, h_toc, level_masl FROM w_levels ORDER BY obsid, date_time' )) reference_string = '(True, [(rb1, 2005-01-01 00:00:00, 222.0, None, None), (rb2, 2005-01-01 00:00:00, 444.0, None, None), (rb2, 2005-01-02 00:00:00, 555.0, None, 667.0)])' print(str(mock_messagebar.mock_calls)) print(test_string) assert test_string == reference_string
def test_calc_selected(self, mock_selected_obsids): mock_selected_obsids.return_value = ['rb1'] db_utils.sql_alter_db( '''INSERT INTO obs_points (obsid, h_toc) VALUES ('rb1', 1)''') db_utils.sql_alter_db( '''INSERT into w_levels (obsid, meas, date_time) VALUES ('rb1', 222, '2005-01-01 00:00:00')''' ) db_utils.sql_alter_db( '''INSERT INTO obs_points (obsid, h_toc) VALUES ('rb2', 4)''') db_utils.sql_alter_db( '''INSERT into w_levels (obsid, meas, date_time) VALUES ('rb2', 444, '2005-01-01 00:00:00')''' ) self.calclvl.FromDateTime = QtWidgets.QDateTimeEdit() self.calclvl.FromDateTime.setDateTime( datestring_to_date('2000-01-01 00:00:00')) self.calclvl.ToDateTime = QtWidgets.QDateTimeEdit() self.calclvl.ToDateTime.setDateTime( datestring_to_date('2010-01-01 00:00:00')) self.calclvl.calcselected() test_string = utils_for_tests.create_test_string( db_utils.sql_load_fr_db( 'SELECT obsid, date_time, meas, h_toc, level_masl FROM w_levels ORDER BY obsid' )) reference_string = '(True, [(rb1, 2005-01-01 00:00:00, 222.0, 1.0, -221.0), (rb2, 2005-01-01 00:00:00, 444.0, None, None)])' assert test_string == reference_string
def test_calcall(self, mock_messagebar): db_utils.sql_alter_db('''INSERT INTO obs_points (obsid, h_toc) VALUES ('rb1', 1)''') db_utils.sql_alter_db('''INSERT into w_levels (obsid, meas, date_time) VALUES ('rb1', 222, '2005-01-01 00:00:00')''') self.calclvl.FromDateTime = QtWidgets.QDateTimeEdit() self.calclvl.FromDateTime.setDateTime(datestring_to_date('2000-01-01 00:00:00')) self.calclvl.ToDateTime = QtWidgets.QDateTimeEdit() self.calclvl.ToDateTime.setDateTime(datestring_to_date('2010-01-01 00:00:00')) self.calclvl.calcall() test_string = utils_for_tests.create_test_string( db_utils.sql_load_fr_db('SELECT obsid, date_time, meas, h_toc, level_masl FROM w_levels')) reference_string = '(True, [(rb1, 2005-01-01 00:00:00, 222.0, 1.0, -221.0)])' assert test_string == reference_string