import numpy

from pydantic import BaseModel

try:
    from pydantic import ConfigDict, RootModel
    PYDANTIC_VERSION = "v2"
except ImportError:
    PYDANTIC_VERSION = "v1"


def test_arbitrary_root():
    if PYDANTIC_VERSION == 'v2':
        # Now it raises suggesting using arbitrary_types_allowed. But it is set!
        # Exception:
        # pydantic.errors.PydanticSchemaGenerationError: Unable to generate pydantic-core schema for <class 'numpy.ndarray'>.
        # Set `arbitrary_types_allowed=True` in the model_config to ignore this error or implement `__get_pydantic_core_schema__` on your type to fully support it.
        class MyNdarray(RootModel[numpy.ndarray]):
            model_config = ConfigDict(arbitrary_types_allowed=True)
            # defines deserialization of list to ndarray etc...
    else:
        # In v1 it works fine.
        class MyNdarray(BaseModel):
            __root__: numpy.ndarray

            class Config:
                arbitrary_types_allowed = True

            # defines deserialization of list to ndarray etc...