Fix test: return Self

Author: dandavison

temporalio/converter.py

@@ -342,6 +342,9 @@ def __init__(self, *converters: EncodingPayloadConverter) -> None:
         Args:
             converters: Payload converters to delegate to, in order.
         """
+        self._set_converters(*converters)
+
+    def _set_converters(self, *converters: EncodingPayloadConverter) -> None:
         self.converters = {c.encoding.encode(): c for c in converters}
 
     def to_payloads(
@@ -406,16 +409,20 @@ def from_payloads(
                 ) from err
         return values
 
-    def with_context(self, context: SerializationContext) -> CompositePayloadConverter:
+    def with_context(self, context: SerializationContext) -> Self:
         """Return a new instance with context set on the component converters"""
-        return CompositePayloadConverter(
+        new_instance = type(self)()
+        new_instance._set_converters(
             *(
-                c.with_context(context)
-                if isinstance(c, WithSerializationContext)
-                else c
+                (
+                    c.with_context(context)
+                    if isinstance(c, WithSerializationContext)
+                    else c
+                )
                 for c in self.converters.values()
             )
         )
+        return new_instance
 
 
 class DefaultPayloadConverter(CompositePayloadConverter):