From c907960a2aa7f94c12478878ffbc48467a0566e9 Mon Sep 17 00:00:00 2001 From: Matthew Rocklin Date: Thu, 19 Apr 2012 10:07:50 -0500 Subject: [PATCH] pretty printing of TransformationSet --- sympy/printing/pretty/pretty.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/sympy/printing/pretty/pretty.py b/sympy/printing/pretty/pretty.py index 7f190d61fcd0..31c9082d15d1 100644 --- a/sympy/printing/pretty/pretty.py +++ b/sympy/printing/pretty/pretty.py @@ -1172,6 +1172,15 @@ def _print_Union(self, u): return self._print_seq(u.args, None, None, union_delimiter, parenthesize = lambda set:set.is_ProductSet or set.is_Intersection) + def _print_TransformationSet(self, ts): + variables = self._print_seq(ts.lamda.variables) + expr = self._print(ts.lamda.expr) + bar = self._print("|") + inn = self._print("in") + base = self._print(ts.base_set) + + return self._print_seq((expr, bar, variables, inn, base), "{", "}", ' ') + def _print_seq(self, seq, left=None, right=None, delimiter=', ', parenthesize=lambda x: False): s = None