def load_test_data(): test_data = News() test_data.set_target_names(config.TARGET_NAMES) data = [] targets = [] with open(TEST_PATH, encoding='utf-8') as f: list_news = f.readlines() for news in list_news: if news and len(news) > 100: news = news.strip() data.append(news[:-2]) targets.append(int(news[-1:])) test_data.set_target(targets) test_data.set_data(data) return test_data