test: add benchmarck core-pydantic

Author: JeanArhancet

src/validators/uuid.rs

@@ -139,18 +139,22 @@ impl Validator for UuidValidator {
 
 impl UuidValidator {
     fn get_uuid<'s, 'data>(&'s self, py: Python<'data>, input: &'data impl Input<'data>) -> ValResult<'data, Uuid> {
-        let uuid = if let Ok(either_string) = input.exact_str() {
-            let cow = either_string.as_cow()?;
-            let uuid_str = cow.as_ref();
-            Uuid::parse_str(uuid_str)
-                .map_err(|e| ValError::new(ErrorType::UuidParsing { error: e.to_string() }, input))?
-        } else if let Ok(either_bytes) = input.validate_bytes(true) {
-            let py_object = either_bytes.into_py(py);
-            let py_bytes = py_object.downcast::<PyBytes>(py)?;
-            Uuid::from_slice(py_bytes.as_bytes())
-                .map_err(|e| ValError::new(ErrorType::UuidParsing { error: e.to_string() }, input))?
-        } else {
-            return Err(ValError::new(ErrorType::UuidType, input));
+        let uuid = match input.exact_str().ok() {
+            Some(either_string) => {
+                let cow = either_string.as_cow()?;
+                let uuid_str = cow.as_ref();
+                Uuid::parse_str(uuid_str)
+                    .map_err(|e| ValError::new(ErrorType::UuidParsing { error: e.to_string() }, input))?
+            }
+            None => {
+                let either_bytes = input
+                    .validate_bytes(true)
+                    .map_err(|_| ValError::new(ErrorType::UuidType, input))?;
+                let py_object = either_bytes.into_py(py);
+                let py_bytes = py_object.downcast::<PyBytes>(py)?;
+                Uuid::from_slice(py_bytes.as_bytes())
+                    .map_err(|e| ValError::new(ErrorType::UuidParsing { error: e.to_string() }, input))?
+            }
         };
 
         if let Some(expected_version) = self.version {

tests/benchmarks/test_micro_benchmarks.py

@@ -687,6 +687,48 @@ class CoreModel:
             }
         )
 
+    @pytest.fixture(scope='class')
+    def validator(self):
+        return SchemaValidator({'type': 'uuid'})
+
+    @pytest.fixture(scope='class')
+    def pydantic_validator(self):
+        def to_UUID(v: Any) -> UUID:
+            try:
+                return UUID(v)
+            except Exception as e:
+                raise PydanticCustomError('uuid_parsing', 'Input should be a valid uuid') from e
+
+        json_schema = core_schema.no_info_after_validator_function(
+            to_UUID, core_schema.str_schema(strict=True, strip_whitespace=True)
+        )
+        schema = core_schema.json_or_python_schema(
+            json_schema=json_schema,
+            python_schema=core_schema.lax_or_strict_schema(
+                lax_schema=core_schema.union_schema([core_schema.is_instance_schema(UUID), json_schema]),
+                strict_schema=core_schema.is_instance_schema(UUID),
+            ),
+            serialization=core_schema.to_string_ser_schema(when_used='json'),
+        )
+
+        return SchemaValidator(schema)
+
+    @pytest.mark.benchmark(group='uuid from str')
+    def test_uuid_from_string_core(self, benchmark, validator):
+        benchmark(validator.validate_python, '12345678-1234-5678-1234-567812345678')
+
+    @pytest.mark.benchmark(group='uuid from str')
+    def test_uuid_from_string_pyd(self, benchmark, pydantic_validator):
+        benchmark(pydantic_validator.validate_python, '12345678-1234-5678-1234-567812345678')
+
+    @pytest.mark.benchmark(group='uuid from UUID')
+    def test_uuid_from_uuid_core(self, benchmark, validator):
+        benchmark(validator.validate_python, UUID('12345678-1234-5678-1234-567812345678'))
+
+    @pytest.mark.benchmark(group='uuid from UUID')
+    def test_uuid_from_uuid_pyd(self, benchmark, pydantic_validator):
+        benchmark(pydantic_validator.validate_python, UUID('12345678-1234-5678-1234-567812345678'))
+
     @pytest.fixture(scope='class')
     def uuid_raw(self):
         return UUID('12345678-1234-5678-1234-567812345678')