from syrupy.extensions.amber import AmberDataSerializer, AmberSnapshotExtension
from syrupy.extensions.amber.pydantic_plugin import PydanticPlugin


class _PydanticAmberDataSerializer(AmberDataSerializer):
    serializer_plugins = [PydanticPlugin]


class PydanticAmberSnapshotExtension(AmberSnapshotExtension):
    serializer_class = _PydanticAmberDataSerializer