def get_test_labelled_rows(): from hardcoded import init_config_yaml, init_raw_csv_file, init_main_yaml RAW_FILE = init_raw_csv_file() SPEC_FILE = init_main_yaml() CFG_FILE = init_config_yaml() return list(get_labelled_rows(RAW_FILE, spec_file = SPEC_FILE, cfg_file = CFG_FILE))
def test_label_csv1(): from hardcoded import init_raw_csv_file, init_main_yaml, PARSED_RAW_FILE_AS_LIST raw_data_file = init_raw_csv_file() SPEC_FILE = init_main_yaml() labelled_rows_as_list = get_labelled_rows_no_segments(raw_data_file, SPEC_FILE) assert labelled_rows_as_list == PARSED_RAW_FILE_AS_LIST