optuna-dashboard/python_tests/test_serializers.py

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"], "<binary object>")

    def test_serialize_string(self) -> None:
        for length in [1000, 1024, 1100]:
            with self.subTest(f"length: {length}"):
                value = "a" * length
                serialized = serialize_attrs(
                    {
                        "key": value,
                    }
                )
                self.assertLessEqual(len(serialized[0]["value"]), 1024)

    def test_serialize_dict(self) -> None:
        serialized = serialize_attrs(
            {
                "key": {"foo": "bar"},
            }
        )
        self.assertLessEqual(len(serialized), 1)