1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
|
"""
PYTEST_DONT_REWRITE
"""
import difflib
import pprint
import pytest
from dirty_equals import HasRepr
from pydantic import BaseModel, ValidationError, field_validator
def _pformat_lines(obj):
return pprint.pformat(obj).splitlines(keepends=True)
def _assert_eq(left, right):
if left != right:
pytest.fail('\n' + '\n'.join(difflib.ndiff(_pformat_lines(left), _pformat_lines(right))))
def test_assert_raises_validation_error():
class Model(BaseModel):
a: str
@field_validator('a')
@classmethod
def check_a(cls, v):
assert v == 'a', 'invalid a'
return v
assert Model(a='a').a == 'a'
with pytest.raises(ValidationError) as exc_info:
Model(a='snap')
_assert_eq(
[
{
'ctx': {'error': HasRepr(repr(AssertionError('invalid a')))},
'input': 'snap',
'loc': ('a',),
'msg': 'Assertion failed, invalid a',
'type': 'assertion_error',
}
],
exc_info.value.errors(include_url=False),
)
|
