Add more __str__ methods to Impl classes

1 files changed, 15 insertions, 0 deletions

diff --git a/polymatrix/expression/impl.py b/polymatrix/expression/impl.py
index 764cab8..81a624e 100644
--- a/polymatrix/expression/impl.py
+++ b/polymatrix/expression/impl.py
@@ -105,6 +105,9 @@ class ARangeExprImpl(ARangeExprMixin):
 class BlockDiagExprImpl(BlockDiagExprMixin):
     blocks: tuple[ExpressionBaseMixin, ...]
 
+    def __str__(self):
+        return f"blkdiag({self.underlying})"
+
 
 @dataclassabc.dataclassabc(frozen=True)
 class CacheExprImpl(CacheExprMixin):
@@ -153,6 +156,9 @@ class DerivativeExprImpl(DerivativeExprMixin):
 class DiagExprImpl(DiagExprMixin):
     underlying: ExpressionBaseMixin
 
+    def __str__(self):
+        return f"diag({self.underlying})"
+
 
 @dataclassabc.dataclassabc(frozen=True)
 class DivergenceExprImpl(DivergenceExprMixin):
@@ -174,6 +180,9 @@ class EvalExprImpl(EvalExprMixin):
     underlying: ExpressionBaseMixin
     substitutions: tuple
 
+    def __str__(self):
+        return f"eval({self.underlying}, {self.substitutions})"
+
 
 @dataclassabc.dataclassabc(frozen=True)
 class EyeExprImpl(EyeExprMixin):
@@ -278,6 +287,9 @@ class LinearMonomialsExprImpl(LinearMonomialsExprMixin):
 class LowerTriangularExprImpl(LowerTriangularExprMixin):
     underlying: ExpressionBaseMixin
 
+    def __str__(self):
+        return f"tril({self.underlying})"
+
 
 @dataclassabc.dataclassabc(frozen=True)
 class LegendreSeriesImpl(LegendreSeriesMixin):
@@ -437,6 +449,9 @@ class SliceExprImpl(SliceExprMixin):
     underlying: ExpressionBaseMixin
     slice: tuple # See SlicePolyMatrixMixin 
 
+    def __str__(self):
+        return f"{self.underlying}[{self.slice}]"
+
 
 @dataclassabc.dataclassabc(frozen=True)
 class SqueezeExprImpl(SqueezeExprMixin):