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

Replace assertEqual with == because we use pytest as a runner

Browse Source
This commit is contained in:
nabenabe0928
2023-11-27 07:33:38 +01:00
parent 6838dd27da
commit ca98208cca
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
@@ -291,7 +291,7 @@ class APITestCase(TestCase):
content_type="application/json",
body=json.dumps(request_body),
)
self.assertEqual(status, 204)
assert status == 204
# Check if the version 1 is deleted.
assert study.system_attrs == {
note_ver_key(0): request_body["version"],
@@ -301,7 +301,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._save_trial_note(request_body)
self.assertEqual(status, 204)
assert status == 204
expected_system_attrs = {
note_ver_key(0): request_body["version"],
f"{note_str_key_prefix(0)}{0}": request_body["body"],
@@ -313,12 +313,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._save_trial_note(request_body)
self.assertEqual(status, 409)
assert status == 409
assert note_ver_key(0) not in study.system_attrs
def test_save_trial_note_empty(self) -> None:
status, study = self._save_trial_note(request_body={})
self.assertEqual(status, 400)
assert status == 400
assert note_ver_key(0) not in study.system_attrs
@pytest.mark.skipif(sys.version_info < (3, 8), reason="BoTorch dropped Python3.7 support")