Skip to content

Commit 29c4264

Browse files
authored
Merge pull request #371 from eendebakpt/fix/init_printing
Fix init_printing
2 parents 5d49ad3 + d1840da commit 29c4264

File tree

2 files changed

+10
-5
lines changed

2 files changed

+10
-5
lines changed

symengine/printing.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,4 @@
1-
from symengine.lib.symengine_wrapper import ccode, sympify, Basic
2-
import symengine.lib.symengine_wrapper
1+
from .lib.symengine_wrapper import ccode, sympify, Basic, repr_latex as _repr_latex
32

43
class CCodePrinter:
54

@@ -29,6 +28,6 @@ def init_printing(pretty_print=True, use_latex=True):
2928
if pretty_print:
3029
if not use_latex:
3130
raise RuntimeError("Only latex is supported for pretty printing")
32-
symengine.lib.symengine_wrapper.repr_latex[0] = True
31+
_repr_latex[0] = True
3332
else:
34-
symengine.lib.symengine_wrapper.repr_latex[0] = False
33+
_repr_latex[0] = False

symengine/tests/test_printing.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
from symengine import (ccode, Symbol, sqrt, Pow, Max, sin, Integer, MutableDenseMatrix)
22
from symengine.utilities import raises
3-
from symengine.printing import CCodePrinter
3+
from symengine.printing import CCodePrinter, init_printing
44

55
def test_ccode():
66
x = Symbol("x")
@@ -24,3 +24,9 @@ def test_CCodePrinter():
2424
assert myprinter.doprint(MutableDenseMatrix(1, 2, [x, y]), "larry") == "larry[0] = x;\nlarry[1] = y;"
2525
raises(TypeError, lambda: myprinter.doprint(sin(x), Integer))
2626
raises(RuntimeError, lambda: myprinter.doprint(MutableDenseMatrix(1, 2, [x, y])))
27+
28+
def test_init_printing():
29+
x = Symbol("x")
30+
assert x._repr_latex_() is None
31+
init_printing()
32+
assert x._repr_latex_() == '$x$'

0 commit comments

Comments
 (0)