allow SymmetricFunctions as arguments of functorial composition

sagemath · Sep 1, 2022 · b2ef1e4 · b2ef1e4
1 parent 8ca3624
commit b2ef1e4
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/src/sage/rings/lazy_series.py b/src/sage/rings/lazy_series.py
@@ -4767,7 +4767,7 @@ def functorial_composition(self, *args):
 
             sage: h = SymmetricFunctions(QQ).h()
             sage: L = LazySymmetricFunctions(h)
-            sage: L(h[2,1]).functorial_composition(L([3*h[0]]))
+            sage: L(h[2,1]).functorial_composition(3*h[0])
             3*h[] + O^7
 
         Check an instance of a non-group action::
@@ -4776,7 +4776,7 @@ def functorial_composition(self, *args):
             sage: p = SymmetricFunctions(QQ).p()
             sage: L = LazySymmetricFunctions(p)
             sage: f = L(lambda n: s[n])
-            sage: g = L(2*s[2, 1, 1] + s[2, 2] + 3*s[4])
+            sage: g = 2*s[2, 1, 1] + s[2, 2] + 3*s[4]
             sage: r = f.functorial_composition(g); r[4]
             Traceback (most recent call last):
             ...
@@ -4793,7 +4793,14 @@ def functorial_composition(self, *args):
         if len(args) == 1:
             g = args[0]
             P = g.parent()
-            R = P._laurent_poly_ring
+            if isinstance(g, LazySymmetricFunction):
+                R = P._laurent_poly_ring
+            else:
+                from sage.rings.lazy_series_ring import LazySymmetricFunctions
+                R = g.parent()
+                P = LazySymmetricFunctions(R)
+                g = P(g)
+
             p = R.realization_of().p()
             # TODO: does the following introduce a memory leak?
             g = Stream_map_coefficients(g._coeff_stream, p)