From d1872b83efebed365e9e389297a65ff261569069 Mon Sep 17 00:00:00 2001 From: keisuke-umezawa Date: Tue, 13 Jun 2023 23:27:38 +0900 Subject: [PATCH] Fix mypy --- e2e_tests/visual_regression_test.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/e2e_tests/visual_regression_test.py b/e2e_tests/visual_regression_test.py index 61e4d2db..484603f1 100644 --- a/e2e_tests/visual_regression_test.py +++ b/e2e_tests/visual_regression_test.py @@ -1,6 +1,3 @@ -import os -import pytest - import optuna from playwright.sync_api import Browser @@ -24,5 +21,10 @@ def test_study_list( for study_id, study_name in study_ids.items(): page.goto(url) page.click(f"a[href='/dashboard/studies/{study_id}']") - title = page.query_selector('.MuiTypography-body1').text_content() - assert study_name in title \ No newline at end of file + + element = page.query_selector(".MuiTypography-body1") + assert element is not None + + title = element.text_content() + assert title is not None + assert study_name in title