Fix tests

Author: Viicos

python/pydantic_core/core_schema.py

@@ -4251,6 +4251,7 @@ def definition_reference_schema(
     'value_error',
     'assertion_error',
     'literal_error',
+    'unset_sentinel_error',
     'date_type',
     'date_parsing',
     'date_from_datetime_parsing',

tests/serializers/test_model.py

@@ -744,6 +744,34 @@ def volume(self) -> int:
     assert s.to_json(Model(3, 4), by_alias=True) == b'{"width":3,"height":4,"Area":12,"volume":48}'
 
 
+def test_computed_field_without_fields() -> None:
+    """https://github.com/pydantic/pydantic/issues/5551"""
+
+    # Original test introduced in https://github.com/pydantic/pydantic-core/pull/550
+
+    class A:
+        @property
+        def b(self) -> str:
+            return 'b'
+
+    schema = core_schema.model_schema(
+        cls=A,
+        config={},
+        schema=core_schema.model_fields_schema(
+            fields={},
+            computed_fields=[
+                core_schema.computed_field('b', return_schema=core_schema.any_schema()),
+            ],
+        ),
+    )
+
+    a = A()
+
+    serializer = SchemaSerializer(schema)
+
+    assert serializer.to_json(a) == b'{"b":"b"}'
+
+
 def test_computed_field_exclude_none():
     @dataclasses.dataclass
     class Model:

tests/test.rs

@@ -53,65 +53,6 @@ mod tests {
         });
     }
 
-    #[test]
-    fn test_serialize_computed_fields() {
-        Python::with_gil(|py| {
-            let code = c_str!(
-                r#"
-class A:
-    @property
-    def b(self) -> str:
-        return "b"
-
-schema = {
-    "cls": A,
-    "config": {},
-    "schema": {
-        "computed_fields": [
-            {"property_name": "b", "return_schema": {"type": "any"}, "type": "computed-field"}
-        ],
-        "fields": {},
-        "type": "model-fields",
-    },
-    "type": "model",
-}
-a = A()
-            "#
-            );
-            let locals = PyDict::new(py);
-            py.run(code, None, Some(&locals)).unwrap();
-            let a = locals.get_item("a").unwrap().unwrap();
-            let schema = locals
-                .get_item("schema")
-                .unwrap()
-                .unwrap()
-                .downcast_into::<PyDict>()
-                .unwrap();
-            let serialized = SchemaSerializer::py_new(schema, None)
-                .unwrap()
-                .to_json(
-                    py,
-                    &a,
-                    None,
-                    Some(false),
-                    None,
-                    None,
-                    Some(true),
-                    false,
-                    false,
-                    false,
-                    false,
-                    WarningsArg::Bool(true),
-                    None,
-                    false,
-                    None,
-                )
-                .unwrap();
-            let serialized: &[u8] = serialized.extract(py).unwrap();
-            assert_eq!(serialized, b"{\"b\":\"b\"}");
-        });
-    }
-
     #[test]
     fn test_literal_schema() {
         Python::with_gil(|py| {

tests/test_errors.py

@@ -340,6 +340,7 @@ def f(input_value, info):
     ('assertion_error', 'Assertion failed, foobar', {'error': AssertionError('foobar')}),
     ('literal_error', 'Input should be foo', {'expected': 'foo'}),
     ('literal_error', 'Input should be foo or bar', {'expected': 'foo or bar'}),
+    ('unset_sentinel_error', "Input should be the 'UNSET' sentinel", None),
     ('date_type', 'Input should be a valid date', None),
     ('date_parsing', 'Input should be a valid date in the format YYYY-MM-DD, foobar', {'error': 'foobar'}),
     ('date_from_datetime_parsing', 'Input should be a valid date or datetime, foobar', {'error': 'foobar'}),

tests/test_schema_functions.py

@@ -82,6 +82,7 @@ def args(*args, **kwargs):
         {'type': 'timedelta', 'microseconds_precision': 'error'},
     ),
     (core_schema.literal_schema, args(['a', 'b']), {'type': 'literal', 'expected': ['a', 'b']}),
+    (core_schema.unset_sentinel_schema, args(), {'type': 'unset-sentinel'}),
     (
         core_schema.enum_schema,
         args(MyEnum, list(MyEnum.__members__.values())),