feat: add benchmarks

Author: JeanArhancet

tests/benchmarks/complete_schema.py

@@ -52,6 +52,7 @@ def wrap_function(input_value, validator, info):
                     'type': 'model-field',
                     'schema': {'type': 'datetime', 'ge': '2000-01-01T06:00:00', 'lt': '2020-01-02T12:13:14'},
                 },
+                'field_uuid': {'type': 'model-field', 'schema': {'type': 'uuid'}},
                 'field_list_any': {'type': 'model-field', 'schema': {'type': 'list'}},
                 'field_list_str': {'type': 'model-field', 'schema': {'type': 'list', 'items_schema': {'type': 'str'}}},
                 'field_list_str_con': {
@@ -226,6 +227,7 @@ def input_data_lax():
         'field_time_con': '12:00:00',
         'field_datetime': '2020-01-01T12:13:14',
         'field_datetime_con': '2020-01-01T00:00:00',
+        'field_uuid': '12345678-1234-5678-1234-567812345678',
         'field_list_any': ['a', b'b', True, 1.0, None] * 10,
         'field_list_str': ['a', 'b', 'c'] * 10,
         'field_list_str_con': ['a', 'b', 'c'] * 10,
@@ -263,6 +265,7 @@ def input_data_lax():
 
 def input_data_strict():
     from datetime import date, datetime, time
+    from uuid import UUID
 
     input_data = input_data_lax()
     input_data.update(
@@ -272,6 +275,7 @@ def input_data_strict():
         field_time_con=time(12, 0, 0),
         field_datetime=datetime(2020, 1, 1, 12, 13, 14),
         field_datetime_con=datetime(2020, 1, 1),
+        field_uuid=UUID('12345678-1234-5678-1234-567812345678'),
     )
     return input_data
 
@@ -293,6 +297,7 @@ def input_data_wrong():
         'field_time_con': '23:00:00',
         'field_datetime': b'smash',
         'field_datetime_con': '1900-01-01T00:00:00',
+        'field_uuid': '12345678-1234-5678-1234-567812345678',
         'field_list_any': {1: 2, 3: 4},
         'field_list_str': [(i,) for i in range(100)],
         'field_list_str_con': ['a', 'b'],

tests/benchmarks/test_complete_benchmark.py

@@ -3,6 +3,7 @@
 """
 import json
 from datetime import date, datetime, time
+from uuid import UUID
 
 import pytest
 
@@ -17,7 +18,7 @@ def test_complete_valid():
     lax_validator = SchemaValidator(lax_schema)
     output = lax_validator.validate_python(input_data_lax())
     assert isinstance(output, cls)
-    assert len(output.__pydantic_fields_set__) == 39
+    assert len(output.__pydantic_fields_set__) == 40
     output_dict = output.__dict__
     assert output_dict == {
         'field_str': 'fo',
@@ -35,6 +36,7 @@ def test_complete_valid():
         'field_time_con': time(12, 0),
         'field_datetime': datetime(2020, 1, 1, 12, 13, 14),
         'field_datetime_con': datetime(2020, 1, 1),
+        'field_uuid': UUID('12345678-1234-5678-1234-567812345678'),
         'field_list_any': ['a', b'b', True, 1.0, None] * 10,
         'field_list_str': ['a', 'b', 'c'] * 10,
         'field_list_str_con': ['a', 'b', 'c'] * 10,

tests/benchmarks/test_micro_benchmarks.py

@@ -8,6 +8,7 @@
 from decimal import Decimal
 from enum import Enum
 from typing import Any, List
+from uuid import UUID
 
 import pytest
 from dirty_equals import IsStr
@@ -669,6 +670,60 @@ def validate_with_expected_error():
         benchmark(validate_with_expected_error)
 
 
+class TestBenchmarkUUID:
+    @pytest.fixture(scope='class')
+    def core_validator(self):
+        class CoreModel:
+            __slots__ = '__dict__', '__pydantic_fields_set__', '__pydantic_extra__', '__pydantic_private__'
+
+        return SchemaValidator(
+            {
+                'type': 'model',
+                'cls': CoreModel,
+                'schema': {
+                    'type': 'model-fields',
+                    'fields': {'u': {'type': 'model-field', 'schema': {'type': 'uuid'}}},
+                },
+            }
+        )
+
+    @pytest.fixture(scope='class')
+    def uuid_raw(self):
+        return UUID('12345678-1234-5678-1234-567812345678')
+
+    @pytest.fixture(scope='class')
+    def uuid_str(self, uuid_raw):
+        return str(uuid_raw)
+
+    @pytest.fixture(scope='class')
+    def python_data_dict(self, uuid_raw):
+        return {'u': uuid_raw}
+
+    @pytest.fixture(scope='class')
+    def json_dict_data(self, uuid_str):
+        return json.dumps({'u': uuid_str})
+
+    @pytest.mark.benchmark(group='uuid model - python')
+    def test_core_python(self, core_validator, benchmark, python_data_dict):
+        benchmark(core_validator.validate_python, python_data_dict)
+
+    @pytest.mark.benchmark(group='uuid model - JSON')
+    def test_model_core_json(self, core_validator, benchmark, json_dict_data):
+        benchmark(core_validator.validate_json, json_dict_data)
+
+    @pytest.mark.benchmark(group='uuid uuid')
+    def test_core_raw(self, benchmark, uuid_raw):
+        v = SchemaValidator({'type': 'uuid'})
+
+        benchmark(v.validate_python, uuid_raw)
+
+    @pytest.mark.benchmark(group='uuid str')
+    def test_core_str(self, benchmark, uuid_str):
+        v = SchemaValidator({'type': 'uuid'})
+
+        benchmark(v.validate_python, uuid_str)
+
+
 @pytest.mark.benchmark(group='raise-error')
 def test_dont_raise_error(benchmark):
     def f(input_value, info):

tests/benchmarks/test_serialization_micro.py

@@ -1,5 +1,6 @@
 import json
 from datetime import date, datetime
+from uuid import UUID
 
 import pytest
 
@@ -358,6 +359,17 @@ def r():
         v.to_python(d, mode='json')
 
 
+@pytest.mark.benchmark(group='model-list-json')
+def test_uuid(benchmark):
+    v = SchemaSerializer(core_schema.uuid_schema())
+    u = UUID('12345678-1234-5678-1234-567812345678')
+    assert v.to_python(u, mode='json') == '12345678-1234-5678-1234-567812345678'
+
+    @benchmark
+    def r():
+        v.to_python(u, mode='json')
+
+
 @pytest.mark.benchmark(group='to-string')
 def test_to_string_format(benchmark):
     s = SchemaSerializer(core_schema.any_schema(serialization=core_schema.format_ser_schema('d')))