diff --git a/python_tests/test_serializers.py b/python_tests/test_serializers.py index 4ea8c9bd..12181538 100644 --- a/python_tests/test_serializers.py +++ b/python_tests/test_serializers.py @@ -5,15 +5,15 @@ from unittest import TestCase from optuna_dashboard._serializer import serialize_attrs -class SerializeAttrsTestCase(TestCase): - def test_serialize_bytes(self) -> None: - serialized = serialize_attrs({"bytes": b"This is a bytes object."}) - self.assertEqual(serialized[0]["value"], "") +def test_serialize_bytes() -> None: + serialized = serialize_attrs({"bytes": b"This is a bytes object."}) + assert serialized[0]["value"] == "" - def test_serialize_dict(self) -> None: - serialized = serialize_attrs( - { - "key": {"foo": "bar"}, - } - ) - self.assertLessEqual(len(serialized), 1) + +def test_serialize_dict() -> None: + serialized = serialize_attrs( + { + "key": {"foo": "bar"}, + } + ) + assert len(serialized) <= 1