def test_read_write_4(self): if os.path.exists('sample.db'): os.unlink('sample.db') database_manager = ExperimentDatabaseManager(file='sample.db', cache_size=40) database_manager.delete_experiment( 'writing_numerical_data_test_case_4') database_manager.set_experiment('writing_numerical_data_test_case_4') inserted_data = dict() inserted_data['var_1'] = 19 inserted_data['var_2'] = 109 inserted_data['var_3'] = np.nan database_manager.insert_experiment_data( 'writing_numerical_data_test_case_4', inserted_data) database_manager.flush() database_manager_2 = ExperimentDatabaseReadingManager(file='sample.db') read_back_data = database_manager_2.get_data( 'writing_numerical_data_test_case_4', 'writing_numerical_data_test_case_4') assert read_back_data['var_1'][0] == inserted_data[ 'var_1'] # Always returns list assert read_back_data['var_2'][0] == inserted_data[ 'var_2'] # Always returns list assert read_back_data['var_3'][0] == 0 # Always returns list # Doing it at the end to check if it flushing works properly database_manager.delete_experiment( 'writing_numerical_data_test_case_4') database_manager.close()
def test_read_write_4(self): database_manager = ExperimentDatabaseManager( mysql_credentials=sql_credentials.credentials, cache_size=40) database_manager.delete_experiment( 'writing_numerical_data_test_case_4') database_manager.set_experiment('writing_numerical_data_test_case_4') inserted_data = dict() inserted_data['var_1'] = 19 inserted_data['var_2'] = 109 inserted_data['var_3'] = np.nan database_manager.insert_experiment_data( 'writing_numerical_data_test_case_4', inserted_data) database_manager.flush() database_manager_2 = ExperimentDatabaseReadingManager( mysql_credentials=sql_credentials.credentials) read_back_data = database_manager_2.get_data( 'writing_numerical_data_test_case_4', 'writing_numerical_data_test_case_4') assert read_back_data['var_1'][0] == inserted_data[ 'var_1'] # Always returns list assert read_back_data['var_2'][0] == inserted_data[ 'var_2'] # Always returns list assert read_back_data['var_3'][0] == 0 # Always returns list # Doing it at the end to check if it flushing works properly database_manager.delete_experiment( 'writing_numerical_data_test_case_4') database_manager.close()
def test_read_write_3(self): if os.path.exists('sample.db'): os.unlink('sample.db') database_manager = ExperimentDatabaseManager(file='sample.db', cache_size=40) database_manager.delete_experiment( 'writing_numerical_data_test_case_3') database_manager.set_experiment('writing_numerical_data_test_case_3') inserted_data = dict() inserted_data['var_1'] = np.array([19, 20]) inserted_data['var_2'] = np.array([109, 110]) inserted_data['var_3'] = np.array([54.1, 43]) inserted_data['var_4'] = np.array(['hello', 'world']) database_manager.insert_experiment_data( 'writing_numerical_data_test_case_3', inserted_data) database_manager.flush() database_manager_2 = ExperimentDatabaseReadingManager(file='sample.db') read_back_data = database_manager_2.get_data( 'writing_numerical_data_test_case_3', 'writing_numerical_data_test_case_3') assert read_back_data['var_1'][0] == inserted_data['var_1'][0] assert read_back_data['var_1'][1] == inserted_data['var_1'][1] assert read_back_data['var_2'][0] == inserted_data['var_2'][0] assert read_back_data['var_2'][1] == inserted_data['var_2'][1] assert read_back_data['var_3'][0] == inserted_data['var_3'][0] assert read_back_data['var_3'][1] == inserted_data['var_3'][1] assert read_back_data['var_4'][0] == inserted_data['var_4'][0] assert read_back_data['var_4'][1] == inserted_data['var_4'][1] # Doing it at the end to check if it flushing works properly # database_manager.delete_experiment('writing_numerical_data_test_case_3') database_manager.close()
def test_read_write_2(self): database_manager = ExperimentDatabaseManager( mysql_credentials=sql_credentials.credentials, cache_size=40) database_manager.delete_experiment( 'writing_numerical_data_test_case_2') database_manager.set_experiment('writing_numerical_data_test_case_2') inserted_data = dict() inserted_data['var_1'] = [19, 20] inserted_data['var_2'] = [109, 110] inserted_data['var_3'] = [54.1, 43] database_manager.insert_experiment_data( 'writing_numerical_data_test_case_2', inserted_data) database_manager.flush() database_manager_2 = ExperimentDatabaseReadingManager( mysql_credentials=sql_credentials.credentials) read_back_data = database_manager_2.get_data( 'writing_numerical_data_test_case_2', 'writing_numerical_data_test_case_2') assert read_back_data['var_1'][0] == inserted_data['var_1'][0] assert read_back_data['var_1'][1] == inserted_data['var_1'][1] assert read_back_data['var_2'][0] == inserted_data['var_2'][0] assert read_back_data['var_2'][1] == inserted_data['var_2'][1] assert read_back_data['var_3'][0] == inserted_data['var_3'][0] assert read_back_data['var_3'][1] == inserted_data['var_3'][1] # Doing it at the end to check if it flushing works properly database_manager.delete_experiment( 'writing_numerical_data_test_case_2') database_manager.close()