Mapytex/mapytex/calculus/random/list.py

"""
List generator
--------------

This function ignores tree structure and works with lists

>>> values = list_generator(["a", "a*b", "b", "c"], conditions=["b%c==1"])
>>> values # doctest: +SKIP
{'a': -8, 'a*b': -40, 'b': 5, 'c': 4}
"""

from .core.generate import random_generator
from .core.grammar import extract_letters, eval_words

def list_generator(var_list, conditions=[], rejected=[0], min_max=(-10, 10), variables_scope={}, dictionnary=False):
    """ Generate random computed values from the list

    :param rd_variables: list of random variables to generate (can be computed value - "a*b")
    :param conditions: condition over variables
    :param rejected  Rejected values for the generator (default [0])
    :param min_max: (min, max) limits in between variables will be generated
    :param variables_scope: rejected and min_max define for individual variables
    :param dictionnary: the return value will be a dictionnary with var_list as keys (default False)
    :return: dictionnary of generated variables

    :example:
    >>> a, ab, b, c = list_generator(["a", "a*b", "b", "c"])
    >>> a, ab, b, c # doctest: +SKIP
    (5, -20, -4, -3)
    >>> a * b == ab
    True
    >>> ab # doctest: +SKIP
    -20
    >>> a, b # doctest: +SKIP
    5, -4
    >>> list_generator(["a", "a*b", "b", "c"], dictionnary=True) # doctest: +SKIP
    {'a': -3, 'a*b': 18, 'b': -6, 'c': -4}
    """
    rv = extract_letters(var_list)
    rv_gen = random_generator(rv, conditions, rejected, min_max, variables_scope)
    generated = eval_words(var_list, rv_gen)
    if dictionnary:
        return generated
    return [generated[v] for v in var_list]