Spaces:
Sleeping
Sleeping
| import pandas as pd | |
| from sheet import (get_serp, get_condensed_result, compose_analysis, | |
| compose_classication, classify_results, extract_results, | |
| postprocess_result) | |
| def test_get_serp(): | |
| # query = "原味商行" | |
| # query = "南投縣中寮鄉中寮村鄉林巷４３號 和興商店" | |
| # query = "啓輝環管企業社" | |
| # query = "蘭陽客棧小吃店" | |
| # query = '韓笑味食品有限公司' | |
| # query = '小阿姨的店' | |
| query = '達米娜魚料理店' | |
| res = get_serp(query, google_domain='google.com.tw') | |
| print(res) | |
| def test_get_condensed_result(): | |
| # query = "原味商行" | |
| # query = "南投縣中寮鄉中寮村鄉林巷４３號 和興商店" | |
| # query = "啓輝環管企業社" | |
| # query = "蘭陽客棧小吃店" | |
| # query = '韓笑味食品有限公司' | |
| # query = '小阿姨的店' | |
| query = '達米娜魚料理店' | |
| res = get_serp(query) | |
| cond_res = get_condensed_result(res) | |
| def test_compose_analysis(): | |
| # query = "原味商行" | |
| # query = "南投縣中寮鄉中寮村鄉林巷４３號 和興商店" | |
| # query = "啓輝環管企業社" | |
| # query = "蘭陽客棧小吃店" | |
| # query = '韓笑味食品有限公司' | |
| # query = '小阿姨的店' | |
| query = '達米娜魚料理店' | |
| res = get_serp(query) | |
| cond_res = get_condensed_result(res) | |
| resp = compose_analysis( query = query, search_results = cond_res) | |
| print( resp ) | |
| def test_compose_classification( evidence): | |
| """ | |
| """ | |
| evidence = '[{"title": "年年有魚餐飲有限公司- 店家介紹", "snippet": "統一編號. 93769370 · 公司狀況. 營業中 · 公司名稱. 年年有魚餐飲有限公司 · 公司類型. 有限公司 · 資本總額. 6000000 · 所在地. 臺中市西區民龍里臺灣大道2段159號1樓."}, {"title": "年年有魚餐飲有限公司", "snippet": "營業地址, 臺中市西區民龍里臺灣大道2段159號1樓 ; 統編, 93769370 ; 營業名稱, 年年有魚餐飲有限公司 ; 資本額, 6,000,000 ; 設立日期, 1120713."}, {"title": "年年有魚餐飲有限公司", "snippet": "公司名稱, 年年有魚餐飲有限公司 ; 資本總額(元), 6,000,000 ; 負責人, 江敏 ; 登記地址, 看地圖 臺中市西區民龍里臺灣大道二段159號1樓 郵遞區號查詢 ; 設立 ..."}, {"title": "年年有魚餐飲有限公司", "snippet": "年年有魚餐飲有限公司 ; 負責人, 江敏 ; 登記地址, 台中市西區民龍里台灣大道二段159號1樓 ; 公司狀態, 核准設立 ; 資本額, 6,000,000元 ; 所在縣市, 台中市 西區 民龍里."}, {"title": "江_敏－年年有魚餐飲有限公司", "snippet": "負責人:江_敏·公司名:年年有魚餐飲有限公司·統一編號:93769370·公司地址:臺中市西區民龍里臺灣大道二段159號1樓·資本額:6000000·公司狀況:核准設立·核准設立 ..."}, {"title": "年年有魚餐飲有限公司/負責人：江_敏", "snippet": "公司名稱:年年有魚餐飲有限公司·代表人姓名:江_敏·公司所在地:臺中市西區民龍里臺灣大道二段159號1樓·統編:93769370資本總額:6000000·公司狀況:核准設立·核准設立 ..."}, {"title": "貓吃魚餐飲有限公司｜工作徵才簡介", "snippet": "貓吃魚餐飲有限公司. 台中市西屯區. 時薪186元. 應徵人數：1 ~ 5人. 排休; 晚班; 工作經驗不拘; 學歷不拘. 1.佈置及清理餐桌2.為顧客帶位或安排座位3.上菜並提供有關用餐的 ..."}, {"title": "食力餐飲_食力國際有限公司｜公司簡介", "snippet": "「食力國際有限公司」正式成立於2023年4月，目前短短時間已成立了四個品牌～ 一、【食力據點】 1:食力咖哩- 台中遠百店（台中市西屯區臺灣大道三段251號大遠百12樓大食 ..."}, {"title": "112 年臺中市優質餐飲店家分級評核獲獎名單", "snippet": "112 年臺中市優質餐飲店家分級評核獲獎名單-. 臺中市餐廳飲食店低碳認證書20 家. 1 築間幸福鍋物-臺中市政二店臺中市西屯區文心路二段213 號. 2 有之和牛-臺中文心店."}, {"title": "年年有魚水族館", "snippet": "營業地址, 臺中市西屯區何安里西屯路2段101-2號1樓 ; 統編, 21833774 ; 營業名稱, 年年有魚水族館 ; 資本額, 60,000 ; 設立日期, 0940502."}, {"title": "類似的店", "snippet": "[\'設計公司\']\\t[\'餐廳\']"}, {"telephone_number": "04 2376 6318"}]' | |
| x = compose_classication( evidence ) | |
| print( x ) | |
| def test_classify_results(): | |
| analysis_results = classify_results( analysis_results) | |
| patch_analysis_results = classify_results( patch_analysis_results) | |
| def test_get_analysis_results(): | |
| data = pd.read_csv("data/餐廳類型分類.xlsx - 測試清單.csv") | |
| analysis_results, empty_indices = extract_results( data ) | |
| def test_postprocess_result(): | |
| analysis_result = "" | |
| pos_res = postprocess_result( analysis_result) |