Add option to change equality sign. Default value is \triangleq

Author: Raphael-CV

src/latexify/__init__.py

@@ -17,3 +17,4 @@
 
 get_latex = core.get_latex
 with_latex = core.with_latex
+set_config = core.set_config

src/latexify/core.py

@@ -28,6 +28,9 @@
 class LatexifyVisitor(node_visitor_base.NodeVisitorBase):
     """Latexify AST visitor."""
 
+    # Default config
+    equality_sign = r"\triangleq"
+
     def __init__(self, math_symbol=False, raw_func_name=False):
         self.math_symbol = math_symbol
         self.raw_func_name = (
@@ -60,7 +63,7 @@ def visit_FunctionDef(self, node, action):  # pylint: disable=invalid-name
             name_str = name_str.replace(r"_", r"\_")  # fix #31
         arg_strs = [self._parse_math_symbols(str(arg.arg)) for arg in node.args.args]
         body_str = self.visit(node.body[0])
-        return name_str + "(" + ", ".join(arg_strs) + r") \triangleq " + body_str
+        return name_str + "(" + ", ".join(arg_strs) + r") " + self.equality_sign + body_str
 
     def visit_Return(self, node, action):  # pylint: disable=invalid-name
         del action
@@ -362,3 +365,12 @@ def ret(fn):
         return _LatexifiedFunction(fn)
 
     return ret
+
+
+def set_config(**kwargs):
+    """
+    Parameters :
+    * `equality_sign`
+    """
+    for key, value in kwargs.items():
+        setattr(LatexifyVisitor, key, value)