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_segment_specs(): from hardcoded import REF_SEGMENT_SPEC, init_config_yaml CFG_FILE = init_config_yaml() segment_specs = _get_segment_specs_no_header_doc(CFG_FILE) assert segment_specs == REF_SEGMENT_SPEC