Update the tests

pydantic · Jan 30, 2024 · 06ee4c9 · 06ee4c9
1 parent b9408e5
commit 06ee4c9
Showing 1 changed file with 19 additions and 2 deletions.
diff --git a/tests/test_types.py b/tests/test_types.py
@@ -4481,11 +4481,28 @@ def test_bytesize_raises():
     class Model(BaseModel):
         size: ByteSize
 
-    with pytest.raises(ValidationError, match='parse value'):
+    with pytest.raises(ValidationError, match='parse value') as exc_info:
         Model(size='d1MB')
+    assert exc_info.value.errors(include_url=False) == [
+        {
+            'input': 'd1MB',
+            'loc': ('size',),
+            'msg': 'could not parse value and unit from byte string',
+            'type': 'byte_size',
+        }
+    ]
 
-    with pytest.raises(ValidationError, match='byte unit'):
+    with pytest.raises(ValidationError, match='byte unit') as exc_info:
         Model(size='1LiB')
+    assert exc_info.value.errors(include_url=False) == [
+        {
+            'ctx': {'unit': 'LiB'},
+            'input': '1LiB',
+            'loc': ('size',),
+            'msg': 'could not interpret byte unit: LiB',
+            'type': 'byte_size_unit',
+        }
+    ]
 
     # 1Gi is not a valid unit unlike 1G
     with pytest.raises(ValidationError, match='byte unit'):