mirror of
https://github.com/wassname/optuna-dashboard.git
synced 2026-09-21 13:10:53 +08:00
Run integration test by using pypputeer
This commit is contained in:
@@ -12,3 +12,4 @@ public/
|
||||
|
||||
*.sqlite3
|
||||
*.db
|
||||
*.png
|
||||
|
||||
@@ -0,0 +1,66 @@
|
||||
import asyncio
|
||||
import optuna
|
||||
import time
|
||||
import threading
|
||||
|
||||
from optuna_dashboard.app import create_app
|
||||
from pyppeteer import launch
|
||||
from pyppeteer.browser import Browser
|
||||
from wsgiref.simple_server import make_server
|
||||
|
||||
|
||||
host = "127.0.0.1"
|
||||
port = 8080
|
||||
|
||||
|
||||
async def take_screenshot(browser: Browser, url: str, path: str):
|
||||
page = await browser.newPage()
|
||||
await page.setViewport({'width': 1200, 'height': 3000})
|
||||
await page.goto(url)
|
||||
time.sleep(5)
|
||||
await page.screenshot({'path': path})
|
||||
|
||||
|
||||
async def integration_test_main():
|
||||
browser = await launch()
|
||||
await take_screenshot(browser, f'http://{host}:{port}/dashboard/', 'study_list.png')
|
||||
await take_screenshot(browser, f'http://{host}:{port}/dashboard/studies/0', 'study_detail_0.png')
|
||||
await take_screenshot(browser, f'http://{host}:{port}/dashboard/studies/1', 'study_detail_1.png')
|
||||
await browser.close()
|
||||
|
||||
|
||||
def run_optuna() -> optuna.storages.InMemoryStorage:
|
||||
storage = optuna.storages.InMemoryStorage()
|
||||
|
||||
# study 0
|
||||
study0 = optuna.create_study(study_name="example-0", storage=storage)
|
||||
|
||||
def objective(trial: optuna.Trial) -> float:
|
||||
x1 = trial.suggest_float("x1", 0, 10)
|
||||
x2 = trial.suggest_float("x2", 0, 10)
|
||||
return (x1 - 2) ** 2 + (x2 - 5) ** 2
|
||||
|
||||
study0.optimize(objective, n_trials=100)
|
||||
|
||||
# study 1 - no trials
|
||||
optuna.create_study(study_name="example-1", storage=storage)
|
||||
return storage
|
||||
|
||||
|
||||
def main():
|
||||
storage = run_optuna()
|
||||
app = create_app(storage)
|
||||
|
||||
httpd = make_server(host, port, app)
|
||||
thread = threading.Thread(target=httpd.serve_forever)
|
||||
thread.start()
|
||||
|
||||
asyncio.get_event_loop().run_until_complete(integration_test_main())
|
||||
|
||||
httpd.shutdown()
|
||||
httpd.server_close()
|
||||
thread.join()
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
Reference in New Issue
Block a user