wassname/optuna-dashboard
1
0
Fork 0
mirror of https://github.com/wassname/optuna-dashboard.git synced 2026-09-09 11:28:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Rename util method in test_save_trial_note

Browse Source
This commit is contained in:
nabenabe0928
2023-11-27 07:31:50 +01:00
parent f29a6e9c0c
commit 6838dd27da
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python_tests/test_api.py
+4 -4
View File
@@ -263,7 +263,7 @@ class APITestCase(TestCase):
self.assertEqual(status, 400)
assert study.trials[0].user_attrs == {}
def _test_save_trial_note(
def _save_trial_note(
self, request_body: dict[str, int | str]
) -> tuple[int, optuna.Study]:
study = optuna.create_study()
@@ -300,7 +300,7 @@ class APITestCase(TestCase):
def test_save_trial_note(self) -> None:
request_body: dict[str, int | str] = {"body": "Test note.", "version": 1}
status, study = self._test_save_trial_note(request_body)
status, study = self._save_trial_note(request_body)
self.assertEqual(status, 204)
expected_system_attrs = {
note_ver_key(0): request_body["version"],
@@ -312,12 +312,12 @@ class APITestCase(TestCase):
def test_save_trial_note_with_wrong_version(self) -> None:
request_body: dict[str, int | str] = {"body": "Test note.", "version": 0}
status, study = self._test_save_trial_note(request_body)
status, study = self._save_trial_note(request_body)
self.assertEqual(status, 409)
assert note_ver_key(0) not in study.system_attrs
def test_save_trial_note_empty(self) -> None:
status, study = self._test_save_trial_note(request_body={})
status, study = self._save_trial_note(request_body={})
self.assertEqual(status, 400)
assert note_ver_key(0) not in study.system_attrs