add doctest in random_expression

This commit is contained in:
lafrite 2014-12-03 15:52:08 +01:00
parent 573945a63a
commit ffd2a55b65

View File

@ -31,6 +31,19 @@ class RdExpression(object):
@classmethod @classmethod
def set_form(cls, form): def set_form(cls, form):
""" Define whether RdExpression create expression with Expression (nice render) or if it only replace inside {} not taking care or render
>>> form = "{a}*{b}"
>>> exp = RdExpression(form)()
>>> print(type(exp))
<class 'pymath.expression.Expression'>
>>> RdExpression.set_form("raw")
>>> form = "{a}*{b}"
>>> exp = RdExpression(form)()
>>> print(type(exp))
<class 'str'>
"""
cls.FORM = form cls.FORM = form
@classmethod @classmethod
@ -189,12 +202,13 @@ if __name__ == '__main__':
rdExp3 = RdExpression(form, cond) rdExp3 = RdExpression(form, cond)
desc_rdExp(rdExp3) desc_rdExp(rdExp3)
form = "{a + a*10}*4 + {a} + 2*{b}" form = "{a+a*10}*4 + {a} + 2*{b}"
cond = ["{a-b} + {b} in list(range(20))", "abs({a}) not in [1]", "{b} not in [1]", "gcd({a},{b}) == 1"] cond = ["{a-b} + {b} in list(range(20))", "abs({a}) not in [1]", "{b} not in [1]", "gcd({a},{b}) == 1"]
rdExp3 = RdExpression(form, cond) rdExp3 = RdExpression(form, cond)
desc_rdExp(rdExp3) desc_rdExp(rdExp3)
import doctest
doctest.testmod()