Upstream pydantic/pydantic-extra-types state of the touched files before 7a922d82eb
This commit is contained in:
upstream committed
1 parent
996db5006c
commit
49b84acf1f
2 files changed
+659
No files matched your search
@@ -0,0 +1,658 @@
|
||||
from typing import Annotated, Any, Union
|
||||
|
||||
import pycountry
|
||||
import pytest
|
||||
from dirty_equals import IsPartialDict
|
||||
from pydantic import BaseModel
|
||||
|
||||
import pydantic_extra_types
|
||||
from pydantic_extra_types import epoch
|
||||
from pydantic_extra_types.color import Color
|
||||
from pydantic_extra_types.coordinate import Coordinate, Latitude, Longitude
|
||||
from pydantic_extra_types.country import CountryAlpha2, CountryAlpha3, CountryNumericCode, CountryShortName
|
||||
from pydantic_extra_types.cron import CronStr
|
||||
from pydantic_extra_types.currency_code import ISO4217, Currency
|
||||
from pydantic_extra_types.domain import DomainStr
|
||||
from pydantic_extra_types.isbn import ISBN
|
||||
from pydantic_extra_types.isin import ISIN
|
||||
from pydantic_extra_types.json_schema import JsonSchema
|
||||
from pydantic_extra_types.language_code import ISO639_3, ISO639_5, LanguageAlpha2, LanguageName
|
||||
from pydantic_extra_types.mac_address import MacAddress
|
||||
from pydantic_extra_types.mime_types import MimeType
|
||||
from pydantic_extra_types.mongo_object_id import MongoObjectId
|
||||
from pydantic_extra_types.payment import PaymentCardNumber
|
||||
from pydantic_extra_types.pendulum_dt import DateTime
|
||||
from pydantic_extra_types.phone_numbers import PhoneNumber, PhoneNumberValidator
|
||||
from pydantic_extra_types.s3 import S3Path
|
||||
from pydantic_extra_types.script_code import ISO_15924
|
||||
from pydantic_extra_types.semantic_version import SemanticVersion
|
||||
from pydantic_extra_types.semver import _VersionPydanticAnnotation
|
||||
from pydantic_extra_types.timezone_name import TimeZoneName
|
||||
from pydantic_extra_types.ulid import ULID
|
||||
|
||||
try:
|
||||
from pydantic_extra_types.pandas import PandasDataFrame, PandasSeries
|
||||
|
||||
pandas_available = True
|
||||
except RuntimeError:
|
||||
pandas_available = False
|
||||
|
||||
languages = [lang.alpha_3 for lang in pycountry.languages]
|
||||
language_families = [lang.alpha_3 for lang in pycountry.language_families]
|
||||
languages.sort()
|
||||
language_families.sort()
|
||||
|
||||
currencies = [currency.alpha_3 for currency in pycountry.currencies]
|
||||
currencies.sort()
|
||||
everyday_currencies = [
|
||||
currency.alpha_3
|
||||
for currency in pycountry.currencies
|
||||
if currency.alpha_3 not in pydantic_extra_types.currency_code._CODES_FOR_BONDS_METAL_TESTING
|
||||
]
|
||||
|
||||
scripts = [script.alpha_4 for script in pycountry.scripts]
|
||||
|
||||
timezone_names = TimeZoneName.allowed_values_list
|
||||
|
||||
everyday_currencies.sort()
|
||||
|
||||
AnyNumberRFC3966 = Annotated[Union[str, PhoneNumber], PhoneNumberValidator()]
|
||||
USNumberE164 = Annotated[
|
||||
Union[str, PhoneNumber],
|
||||
PhoneNumberValidator(
|
||||
supported_regions=['US'],
|
||||
default_region='US',
|
||||
number_format='E164',
|
||||
),
|
||||
]
|
||||
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
'cls,expected',
|
||||
[
|
||||
(
|
||||
Color,
|
||||
{
|
||||
'properties': {'x': {'format': 'color', 'title': 'X', 'type': 'string'}},
|
||||
'required': ['x'],
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
},
|
||||
),
|
||||
(
|
||||
PaymentCardNumber,
|
||||
{
|
||||
'properties': {
|
||||
'x': {
|
||||
'maxLength': 19,
|
||||
'minLength': 12,
|
||||
'title': 'X',
|
||||
'type': 'string',
|
||||
}
|
||||
},
|
||||
'required': ['x'],
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
},
|
||||
),
|
||||
(
|
||||
CountryAlpha2,
|
||||
{
|
||||
'properties': {'x': {'pattern': '^\\w{2}$', 'title': 'X', 'type': 'string'}},
|
||||
'required': ['x'],
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
},
|
||||
),
|
||||
(
|
||||
CronStr,
|
||||
{
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
'properties': {
|
||||
'x': {
|
||||
'title': 'X',
|
||||
'type': 'string',
|
||||
}
|
||||
},
|
||||
'required': ['x'],
|
||||
},
|
||||
),
|
||||
(
|
||||
CountryAlpha3,
|
||||
{
|
||||
'properties': {'x': {'pattern': '^\\w{3}$', 'title': 'X', 'type': 'string'}},
|
||||
'required': ['x'],
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
},
|
||||
),
|
||||
(
|
||||
CountryNumericCode,
|
||||
{
|
||||
'properties': {'x': {'pattern': '^[0-9]{3}$', 'title': 'X', 'type': 'string'}},
|
||||
'required': ['x'],
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
},
|
||||
),
|
||||
(
|
||||
CountryShortName,
|
||||
{
|
||||
'properties': {'x': {'title': 'X', 'type': 'string'}},
|
||||
'required': ['x'],
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
},
|
||||
),
|
||||
(
|
||||
MacAddress,
|
||||
{
|
||||
'properties': {
|
||||
'x': {
|
||||
'title': 'X',
|
||||
'type': 'string',
|
||||
}
|
||||
},
|
||||
'required': ['x'],
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
},
|
||||
),
|
||||
(
|
||||
Latitude,
|
||||
{
|
||||
'properties': {
|
||||
'x': {
|
||||
'anyOf': [
|
||||
{'maximum': 90.0, 'minimum': -90.0, 'type': 'number'},
|
||||
{'type': 'string', 'pattern': '^(?!^[-+.]*$)[+-]?0*\\d*\\.?\\d*$'},
|
||||
],
|
||||
'title': 'X',
|
||||
}
|
||||
},
|
||||
'required': ['x'],
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
},
|
||||
),
|
||||
(
|
||||
Longitude,
|
||||
{
|
||||
'properties': {
|
||||
'x': {
|
||||
'anyOf': [
|
||||
{'maximum': 180.0, 'minimum': -180.0, 'type': 'number'},
|
||||
{'type': 'string', 'pattern': '^(?!^[-+.]*$)[+-]?0*\\d*\\.?\\d*$'},
|
||||
],
|
||||
'title': 'X',
|
||||
}
|
||||
},
|
||||
'required': ['x'],
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
},
|
||||
),
|
||||
(
|
||||
Coordinate,
|
||||
{
|
||||
'$defs': {
|
||||
'Coordinate': IsPartialDict(
|
||||
{
|
||||
'properties': {
|
||||
'latitude': {
|
||||
'anyOf': [
|
||||
{'maximum': 90.0, 'minimum': -90.0, 'type': 'number'},
|
||||
{'type': 'string', 'pattern': '^(?!^[-+.]*$)[+-]?0*\\d*\\.?\\d*$'},
|
||||
],
|
||||
'title': 'Latitude',
|
||||
},
|
||||
'longitude': {
|
||||
'anyOf': [
|
||||
{'maximum': 180.0, 'minimum': -180.0, 'type': 'number'},
|
||||
{'type': 'string', 'pattern': '^(?!^[-+.]*$)[+-]?0*\\d*\\.?\\d*$'},
|
||||
],
|
||||
'title': 'Longitude',
|
||||
},
|
||||
},
|
||||
'required': ['latitude', 'longitude'],
|
||||
'title': 'Coordinate',
|
||||
'type': 'object',
|
||||
},
|
||||
)
|
||||
},
|
||||
'properties': {
|
||||
'x': {
|
||||
'anyOf': [
|
||||
{'$ref': '#/$defs/Coordinate'},
|
||||
{
|
||||
'maxItems': 2,
|
||||
'minItems': 2,
|
||||
'prefixItems': [
|
||||
{
|
||||
'anyOf': [
|
||||
{'type': 'number'},
|
||||
{'type': 'string', 'pattern': '^(?!^[-+.]*$)[+-]?0*\\d*\\.?\\d*$'},
|
||||
]
|
||||
},
|
||||
{
|
||||
'anyOf': [
|
||||
{'type': 'number'},
|
||||
{'type': 'string', 'pattern': '^(?!^[-+.]*$)[+-]?0*\\d*\\.?\\d*$'},
|
||||
]
|
||||
},
|
||||
],
|
||||
'type': 'array',
|
||||
},
|
||||
{'type': 'string'},
|
||||
],
|
||||
'title': 'X',
|
||||
},
|
||||
},
|
||||
'required': ['x'],
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
},
|
||||
),
|
||||
(
|
||||
ULID,
|
||||
{
|
||||
'properties': {
|
||||
'x': {
|
||||
'anyOf': [
|
||||
{'type': 'integer'},
|
||||
{'format': 'binary', 'type': 'string'},
|
||||
{'type': 'string'},
|
||||
{'format': 'uuid', 'type': 'string'},
|
||||
],
|
||||
'title': 'X',
|
||||
}
|
||||
},
|
||||
'required': ['x'],
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
},
|
||||
),
|
||||
(
|
||||
ISBN,
|
||||
{
|
||||
'properties': {
|
||||
'x': {
|
||||
'title': 'X',
|
||||
'type': 'string',
|
||||
}
|
||||
},
|
||||
'required': ['x'],
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
},
|
||||
),
|
||||
(
|
||||
JsonSchema,
|
||||
{
|
||||
'properties': {
|
||||
'x': {
|
||||
'anyOf': [
|
||||
{'type': 'string'},
|
||||
{'additionalProperties': True, 'type': 'object'},
|
||||
],
|
||||
'title': 'X',
|
||||
}
|
||||
},
|
||||
'required': ['x'],
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
},
|
||||
),
|
||||
(
|
||||
ISIN,
|
||||
{
|
||||
'properties': {
|
||||
'x': {
|
||||
'title': 'X',
|
||||
'type': 'string',
|
||||
}
|
||||
},
|
||||
'required': ['x'],
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
},
|
||||
),
|
||||
(
|
||||
DateTime,
|
||||
{
|
||||
'properties': {'x': {'format': 'date-time', 'title': 'X', 'type': 'string'}},
|
||||
'required': ['x'],
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
},
|
||||
),
|
||||
(
|
||||
LanguageAlpha2,
|
||||
{
|
||||
'properties': {'x': {'pattern': '^\\w{2}$', 'title': 'X', 'type': 'string'}},
|
||||
'required': ['x'],
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
},
|
||||
),
|
||||
(
|
||||
LanguageName,
|
||||
{
|
||||
'properties': {'x': {'title': 'X', 'type': 'string'}},
|
||||
'required': ['x'],
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
},
|
||||
),
|
||||
(
|
||||
ISO639_3,
|
||||
{
|
||||
'properties': {
|
||||
'x': {
|
||||
'title': 'X',
|
||||
'type': 'string',
|
||||
'enum': languages,
|
||||
'maxLength': 3,
|
||||
'minLength': 3,
|
||||
}
|
||||
},
|
||||
'required': ['x'],
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
},
|
||||
),
|
||||
(
|
||||
ISO639_5,
|
||||
{
|
||||
'properties': {
|
||||
'x': {
|
||||
'title': 'X',
|
||||
'type': 'string',
|
||||
'enum': language_families,
|
||||
'maxLength': 3,
|
||||
'minLength': 3,
|
||||
}
|
||||
},
|
||||
'required': ['x'],
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
},
|
||||
),
|
||||
(
|
||||
ISO4217,
|
||||
{
|
||||
'properties': {
|
||||
'x': {
|
||||
'title': 'X',
|
||||
'type': 'string',
|
||||
'enum': currencies,
|
||||
'maxLength': 3,
|
||||
'minLength': 3,
|
||||
}
|
||||
},
|
||||
'required': ['x'],
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
},
|
||||
),
|
||||
(
|
||||
Currency,
|
||||
{
|
||||
'properties': {
|
||||
'x': {
|
||||
'title': 'X',
|
||||
'type': 'string',
|
||||
'enum': everyday_currencies,
|
||||
'maxLength': 3,
|
||||
'minLength': 3,
|
||||
}
|
||||
},
|
||||
'required': ['x'],
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
},
|
||||
),
|
||||
(
|
||||
ISO_15924,
|
||||
{
|
||||
'properties': {
|
||||
'x': {
|
||||
'title': 'X',
|
||||
'type': 'string',
|
||||
'enum': scripts,
|
||||
'maxLength': 4,
|
||||
'minLength': 4,
|
||||
}
|
||||
},
|
||||
'required': ['x'],
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
},
|
||||
),
|
||||
(
|
||||
SemanticVersion,
|
||||
{
|
||||
'properties': {
|
||||
'x': {
|
||||
'title': 'X',
|
||||
'type': 'string',
|
||||
'pattern': r'^(0|[1-9]\d*)\.(0|[1-9]\d*)\.(0|[1-9]\d*)(?:-((?:0|[1-9]\d*|\d*[a-zA-Z-][0-9a-zA-Z-]*)(?:\.(?:0|[1-9]\d*|\d*[a-zA-Z-][0-9a-zA-Z-]*))*))?(?:\+([0-9a-zA-Z-]+(?:\.[0-9a-zA-Z-]+)*))?$',
|
||||
}
|
||||
},
|
||||
'required': ['x'],
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
},
|
||||
),
|
||||
(
|
||||
TimeZoneName,
|
||||
{
|
||||
'properties': {
|
||||
'x': {
|
||||
'title': 'X',
|
||||
'type': 'string',
|
||||
'enum': timezone_names,
|
||||
'minLength': 1,
|
||||
}
|
||||
},
|
||||
'required': ['x'],
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
},
|
||||
),
|
||||
(
|
||||
_VersionPydanticAnnotation,
|
||||
{
|
||||
'properties': {'x': {'title': 'X', 'type': 'string'}},
|
||||
'required': ['x'],
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
},
|
||||
),
|
||||
(
|
||||
PhoneNumber,
|
||||
{
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
'properties': {
|
||||
'x': {
|
||||
'title': 'X',
|
||||
'type': 'string',
|
||||
'format': 'phone',
|
||||
}
|
||||
},
|
||||
'required': ['x'],
|
||||
},
|
||||
),
|
||||
(
|
||||
AnyNumberRFC3966,
|
||||
{
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
'properties': {
|
||||
'x': {
|
||||
'title': 'X',
|
||||
'type': 'string',
|
||||
'format': 'phone',
|
||||
}
|
||||
},
|
||||
'required': ['x'],
|
||||
},
|
||||
),
|
||||
(
|
||||
USNumberE164,
|
||||
{
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
'properties': {
|
||||
'x': {
|
||||
'title': 'X',
|
||||
'type': 'string',
|
||||
'format': 'phone',
|
||||
}
|
||||
},
|
||||
'required': ['x'],
|
||||
},
|
||||
),
|
||||
(
|
||||
S3Path,
|
||||
{
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
'properties': {
|
||||
'x': {
|
||||
'pattern': '^s3://([^/]+)/(.*?([^/]+)/?)$',
|
||||
'title': 'X',
|
||||
'type': 'string',
|
||||
},
|
||||
},
|
||||
'required': [
|
||||
'x',
|
||||
],
|
||||
},
|
||||
),
|
||||
(
|
||||
DomainStr,
|
||||
{
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
'properties': {
|
||||
'x': {
|
||||
'title': 'X',
|
||||
'type': 'string',
|
||||
},
|
||||
},
|
||||
'required': [
|
||||
'x',
|
||||
],
|
||||
},
|
||||
),
|
||||
(
|
||||
epoch.Integer,
|
||||
{
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
'properties': {
|
||||
'x': {
|
||||
'title': 'X',
|
||||
'type': 'integer',
|
||||
'format': 'date-time',
|
||||
},
|
||||
},
|
||||
'required': [
|
||||
'x',
|
||||
],
|
||||
},
|
||||
),
|
||||
(
|
||||
epoch.Number,
|
||||
{
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
'properties': {
|
||||
'x': {
|
||||
'title': 'X',
|
||||
'type': 'number',
|
||||
'format': 'date-time',
|
||||
},
|
||||
},
|
||||
'required': [
|
||||
'x',
|
||||
],
|
||||
},
|
||||
),
|
||||
(
|
||||
MongoObjectId,
|
||||
{
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
'properties': {
|
||||
'x': {
|
||||
'maxLength': MongoObjectId.OBJECT_ID_LENGTH,
|
||||
'minLength': MongoObjectId.OBJECT_ID_LENGTH,
|
||||
'title': 'X',
|
||||
'type': 'string',
|
||||
},
|
||||
},
|
||||
'required': ['x'],
|
||||
},
|
||||
),
|
||||
(
|
||||
MimeType,
|
||||
{
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
'properties': {
|
||||
'x': {
|
||||
'pattern': r'^[a-zA-Z0-9][a-zA-Z0-9!#$&\-\^_+.]+/[a-zA-Z0-9][a-zA-Z0-9!#$&\-\^_+.]+$',
|
||||
'title': 'X',
|
||||
'type': 'string',
|
||||
},
|
||||
},
|
||||
'required': ['x'],
|
||||
},
|
||||
),
|
||||
]
|
||||
+ (
|
||||
[
|
||||
(
|
||||
PandasDataFrame,
|
||||
{
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
'properties': {
|
||||
'x': {
|
||||
'title': 'DataFrame',
|
||||
'type': 'object',
|
||||
},
|
||||
},
|
||||
'required': ['x'],
|
||||
},
|
||||
),
|
||||
(
|
||||
PandasSeries,
|
||||
{
|
||||
'title': 'Model',
|
||||
'type': 'object',
|
||||
'properties': {
|
||||
'x': {
|
||||
'title': 'Series',
|
||||
'type': 'array',
|
||||
},
|
||||
},
|
||||
'required': ['x'],
|
||||
},
|
||||
),
|
||||
]
|
||||
if pandas_available
|
||||
else []
|
||||
),
|
||||
)
|
||||
def test_json_schema(cls: Any, expected: dict[str, Any]) -> None:
|
||||
"""Test the model_json_schema implementation for all extra types."""
|
||||
|
||||
class Model(BaseModel):
|
||||
x: cls
|
||||
|
||||
assert Model.model_json_schema() == expected
|
||||
Reference in new issue
Block a user