Thank you! Fixed some whitespaces and pushed it in.

Vinzent

On Sat, Apr 18, 2009 at 05:02:51PM +0200, Fabian Pedregosa wrote:
> 
> That way we get a chance to disable line wrapping, which is interesting, for 
> example
> for pretty printing to non-terminals (html, etc.)
> ---
>  sympy/printing/pretty/pretty.py                    |   18 +++++++++++++-----
>  sympy/printing/pretty/tests/test_pretty_unicode.py |    6 ++++++
>  2 files changed, 19 insertions(+), 5 deletions(-)
> 
> diff --git a/sympy/printing/pretty/pretty.py b/sympy/printing/pretty/pretty.py
> index 9a31e46..afa4dcd 100644
> --- a/sympy/printing/pretty/pretty.py
> +++ b/sympy/printing/pretty/pretty.py
> @@ -14,20 +14,23 @@
>  class PrettyPrinter(Printer):
>      """Printer, which converts an expression into 2D ascii-art figure."""
>  
> -    def __init__(self, profile=None):
> +    def __init__(self, profile=None, wrap=True):
>          Printer.__init__(self)
>          self.emptyPrinter = lambda x : prettyForm(xstr(x))
>  
>          self._settings = {
>                  "full_prec" : "auto",
>                  "use_unicode" : True,
> +                "wrap" : wrap,
>          }
>  
>          if profile is not None:
>              self._settings.update(profile)
>  
>      def doprint(self, expr):
> -        return self._print(expr).terminal_string()
> +        if self._settings["wrap"]:
> +            return self._print(expr).terminal_string()
> +        return unicode( self._print(expr) )
>  
>      # empty op so _print(stringPict) returns the same
>      def _print_stringPict(self, e):
> @@ -591,9 +594,14 @@ def __print_set(self, s):
>  
>  def pretty(expr, profile=None, **kargs):
>      """
> -    Returns a string containing the prettified form of expr. If use_unicode
> -    is set to True then certain expressions will use unicode characters,
> -    such as the greek letter pi for Pi instances.
> +    Returns a string containing the prettified form of expr. 
> +    
> +    Arguments
> +    ---------
> +    expr: the expression to print
> +    wrap: line wrapping enabled/disabled, should be a boolean value (default 
> to True)
> +    use_unicode: use unicode characters, such as the greek letter pi instead 
> of 
> +        the string pi. Values should be boolean or None
>      """
>      if profile is not None:
>          profile.update(kargs)
> diff --git a/sympy/printing/pretty/tests/test_pretty_unicode.py 
> b/sympy/printing/pretty/tests/test_pretty_unicode.py
> index ee7bb16..cccfbd0 100644
> --- a/sympy/printing/pretty/tests/test_pretty_unicode.py
> +++ b/sympy/printing/pretty/tests/test_pretty_unicode.py
> @@ -386,3 +386,9 @@ def test_upretty_seq_even():
>  """
>      assert u == s
>  
> +def test_pretty_nowrap():
> +    huge_expr = 0
> +    for i in range(20):
> +        huge_expr += i*sin(i+x)
> +    assert pretty(huge_expr            ).find('\n') != -1
> +    assert pretty(huge_expr, wrap=False).find('\n') == -1
> -- 
> 1.6.1.2
> 
> 
> > 

--~--~---------~--~----~------------~-------~--~----~
You received this message because you are subscribed to the Google Groups 
"sympy-patches" group.
To post to this group, send email to sympy-patches@googlegroups.com
To unsubscribe from this group, send email to 
sympy-patches+unsubscr...@googlegroups.com
For more options, visit this group at 
http://groups.google.com/group/sympy-patches?hl=en
-~----------~----~----~----~------~----~------~--~---

Reply via email to