pydantic/tests/test_callable.py

from typing import Callable

import pytest

from pydantic import BaseModel, ValidationError


@pytest.mark.parametrize('annotation', [Callable, Callable[[int], int]])
def test_callable(annotation):
    class Model(BaseModel):
        callback: annotation

    m = Model(callback=lambda x: x)
    assert callable(m.callback)


@pytest.mark.parametrize('annotation', [Callable, Callable[[int], int]])
def test_non_callable(annotation):
    class Model(BaseModel):
        callback: annotation

    with pytest.raises(ValidationError):
        Model(callback=1)