Mapytex/mapytex/calculus/random/expression.py

from ..API.expression import Expression
from .core.random_tree import RandomTree

DEFAUTL_CONFIG = {
        "rejected": [0, 1],
        "min_max": (-10, 10),
        }

def expression_generator(template:str, conditions:list[str]=[], global_config:dict={}, configs:dict={}):
    """ Generate a random expression 

    :param template: the template of the expression
    :param conditions: conditions on randomly generate variable
    :param global_config: configuration for all variables
    :param configs: configuration for each variables
    :return: Expression or Token generated

    :example:
    >>> e = expression_generator("{a}/{a*k}")
    >>> e # doctest: +SKIP
    <Exp: -3 / -15>
    >>> e = expression_generator("{a}/{a*k} - 3*{b}", configs={'a':{'min_max':(10, 30)}})
    >>> e # doctest: +SKIP
    <Exp: 18 / 108 - 3 * 9>
    >>> e = expression_generator("{a}*x + {b}*x + 3", conditions=["a>b"], global_config={"rejected":[0, 1]})
    >>> print(e) # doctest: +SKIP
    10x - 6x + 3
    >>> ee = e.simplify()
    >>> print(ee) # doctest: +SKIP
    4x + 3

    """
    rd_tree = RandomTree.from_str(template)
    generated_tree = rd_tree.generate(conditions, dict(DEFAUTL_CONFIG, **global_config), configs)
    return Expression._post_processing(generated_tree)