65 lines
1.8 KiB
Python
65 lines
1.8 KiB
Python
"""
|
|
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
|
|
|
|
DEFAUTL_CONFIG = {
|
|
"rejected": [0],
|
|
"min_max": (-10, 10),
|
|
}
|
|
|
|
|
|
def list_generator(
|
|
var_list: list[str],
|
|
conditions: list[str] = [],
|
|
global_config: dict = {},
|
|
configs: dict = {},
|
|
) -> list[int]:
|
|
"""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 global_config: configuration for all variables
|
|
:param configs: configuration for each variables
|
|
:return: list 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
|
|
>>> a, ab, b, c = list_generator(["a", "a*b", "b", "c"], conditions=["a-b==0"])
|
|
>>> a - b == 0
|
|
True
|
|
>>> a, ab, b, c = list_generator(["a", "a*b", "b", "c"], global_config={"rejected": [2, 3, 5, 7]})
|
|
>>> a not in [2, 3, 5, 7]
|
|
True
|
|
>>> b not in [2, 3, 5, 7]
|
|
True
|
|
>>> c not in [2, 3, 5, 7]
|
|
True
|
|
>>> a, ab, b, c = list_generator(["a", "a*b", "b", "c"], configs={"a": {"rejected": [2, 3, 5, 7]}})
|
|
>>> a not in [2, 3, 5, 7]
|
|
True
|
|
"""
|
|
rv = extract_letters(var_list)
|
|
rv_gen = random_generator(
|
|
rv, conditions, dict(DEFAUTL_CONFIG, **global_config), configs
|
|
)
|
|
generated = eval_words(var_list, rv_gen)
|
|
return [generated[v] for v in var_list]
|