Fix: format with black
This commit is contained in:
parent
a267acd7b3
commit
aad2395a3a
@ -10,8 +10,10 @@
|
|||||||
from random import choice
|
from random import choice
|
||||||
|
|
||||||
|
|
||||||
def complete_variable_configs(variables, global_config:dict={}, configs:dict={})->dict:
|
def complete_variable_configs(
|
||||||
""" Completes variables configurations with the global configuration
|
variables, global_config: dict = {}, configs: dict = {}
|
||||||
|
) -> dict:
|
||||||
|
"""Completes variables configurations with the global configuration
|
||||||
|
|
||||||
:param variables: list of random variables to generate
|
:param variables: list of random variables to generate
|
||||||
:param global_config: global parameters
|
:param global_config: global parameters
|
||||||
@ -47,9 +49,12 @@ def complete_variable_configs(variables, global_config:dict={}, configs:dict={})
|
|||||||
|
|
||||||
|
|
||||||
def random_generator(
|
def random_generator(
|
||||||
variables:list[str], conditions:list[str]=[], global_config:dict={}, configs:dict={},
|
variables: list[str],
|
||||||
)-> dict[str, int]:
|
conditions: list[str] = [],
|
||||||
""" Generate random variables
|
global_config: dict = {},
|
||||||
|
configs: dict = {},
|
||||||
|
) -> dict[str, int]:
|
||||||
|
"""Generate random variables
|
||||||
|
|
||||||
:param variables: list of random variables to generate
|
:param variables: list of random variables to generate
|
||||||
:param conditions: condition over variables
|
:param conditions: condition over variables
|
||||||
@ -81,9 +86,7 @@ def random_generator(
|
|||||||
>>> gene["a"] % gene["b"]
|
>>> gene["a"] % gene["b"]
|
||||||
0
|
0
|
||||||
"""
|
"""
|
||||||
complete_scope = complete_variable_configs(
|
complete_scope = complete_variable_configs(variables, global_config, configs)
|
||||||
variables, global_config, configs
|
|
||||||
)
|
|
||||||
choices_list = {
|
choices_list = {
|
||||||
v: list(
|
v: list(
|
||||||
set(
|
set(
|
||||||
@ -105,5 +108,3 @@ def random_generator(
|
|||||||
generate_variable = {v: choice(choices_list[v]) for v in variables}
|
generate_variable = {v: choice(choices_list[v]) for v in variables}
|
||||||
|
|
||||||
return generate_variable
|
return generate_variable
|
||||||
|
|
||||||
|
|
||||||
|
@ -1,6 +1,5 @@
|
|||||||
|
def extract_letters(words: list[str]) -> set[str]:
|
||||||
def extract_letters(words:list[str])->set[str]:
|
"""Extracts unique letters from a list of words
|
||||||
""" Extracts unique letters from a list of words
|
|
||||||
|
|
||||||
:param words: list of leafs
|
:param words: list of leafs
|
||||||
:return: set of letters
|
:return: set of letters
|
||||||
@ -18,8 +17,8 @@ def extract_letters(words:list[str])->set[str]:
|
|||||||
return letters
|
return letters
|
||||||
|
|
||||||
|
|
||||||
def eval_words(words:list[str], values:dict[str,int]) -> dict[str, int]:
|
def eval_words(words: list[str], values: dict[str, int]) -> dict[str, int]:
|
||||||
""" Evaluate words replacing letters with values
|
"""Evaluate words replacing letters with values
|
||||||
|
|
||||||
:param words: list of words
|
:param words: list of words
|
||||||
:param values: Dictionary of letters:value
|
:param values: Dictionary of letters:value
|
||||||
@ -32,4 +31,3 @@ def eval_words(words:list[str], values:dict[str,int]) -> dict[str, int]:
|
|||||||
{'a': 2, 'a*k': 6}
|
{'a': 2, 'a*k': 6}
|
||||||
"""
|
"""
|
||||||
return {word: eval(word, values) for word in words}
|
return {word: eval(word, values) for word in words}
|
||||||
|
|
||||||
|
@ -16,7 +16,7 @@ __all__ = ["reject_random", "filter_random", "FilterRandom"]
|
|||||||
|
|
||||||
|
|
||||||
def reject_random(min_value=-10, max_value=10, rejected=[0, 1], accept_callbacks=[]):
|
def reject_random(min_value=-10, max_value=10, rejected=[0, 1], accept_callbacks=[]):
|
||||||
""" Generate a random integer with the rejection method
|
"""Generate a random integer with the rejection method
|
||||||
|
|
||||||
:param name: name of the Integer
|
:param name: name of the Integer
|
||||||
:param min_value: minimum value
|
:param min_value: minimum value
|
||||||
@ -60,7 +60,7 @@ def reject_random(min_value=-10, max_value=10, rejected=[0, 1], accept_callbacks
|
|||||||
|
|
||||||
|
|
||||||
def filter_random(min_value=-10, max_value=10, rejected=[0, 1], accept_callbacks=[]):
|
def filter_random(min_value=-10, max_value=10, rejected=[0, 1], accept_callbacks=[]):
|
||||||
""" Generate a random integer by filtering then choosing a candidate
|
"""Generate a random integer by filtering then choosing a candidate
|
||||||
|
|
||||||
:param name: name of the Integer
|
:param name: name of the Integer
|
||||||
:param min_value: minimum value
|
:param min_value: minimum value
|
||||||
@ -111,8 +111,7 @@ def filter_random(min_value=-10, max_value=10, rejected=[0, 1], accept_callbacks
|
|||||||
|
|
||||||
class FilterRandom(object):
|
class FilterRandom(object):
|
||||||
|
|
||||||
""" Integer random generator which filter then choose candidate
|
"""Integer random generator which filter then choose candidate"""
|
||||||
"""
|
|
||||||
|
|
||||||
# TODO: Faire un cache pour éviter de reconstruire les listes à chaque fois |ven. déc. 21 19:07:42 CET 2018
|
# TODO: Faire un cache pour éviter de reconstruire les listes à chaque fois |ven. déc. 21 19:07:42 CET 2018
|
||||||
|
|
||||||
@ -133,7 +132,7 @@ class FilterRandom(object):
|
|||||||
}
|
}
|
||||||
|
|
||||||
def add_candidates(self, low, high):
|
def add_candidates(self, low, high):
|
||||||
""" Add candidates between low and high to _candidates """
|
"""Add candidates between low and high to _candidates"""
|
||||||
if low < self._min:
|
if low < self._min:
|
||||||
self._min = low
|
self._min = low
|
||||||
useless_low = False
|
useless_low = False
|
||||||
@ -157,11 +156,11 @@ class FilterRandom(object):
|
|||||||
)
|
)
|
||||||
|
|
||||||
def candidates(self, min_value=-10, max_value=10):
|
def candidates(self, min_value=-10, max_value=10):
|
||||||
""" Return candidates between min_value and max_value """
|
"""Return candidates between min_value and max_value"""
|
||||||
return [c for c in self._candidates if (c > min_value and c < max_value)]
|
return [c for c in self._candidates if (c > min_value and c < max_value)]
|
||||||
|
|
||||||
def __call__(self, min_value=-10, max_value=10):
|
def __call__(self, min_value=-10, max_value=10):
|
||||||
""" Randomly choose on candidate """
|
"""Randomly choose on candidate"""
|
||||||
self.add_candidates(min_value, max_value)
|
self.add_candidates(min_value, max_value)
|
||||||
return random.choice(self.candidates(min_value, max_value))
|
return random.choice(self.candidates(min_value, max_value))
|
||||||
|
|
||||||
|
@ -14,7 +14,7 @@ from ...core.coroutine import coroutine, STOOOP
|
|||||||
|
|
||||||
@coroutine
|
@coroutine
|
||||||
def look_for_rdleaf(target):
|
def look_for_rdleaf(target):
|
||||||
""" Coroutine which look to "{...}" which are RdLeaf
|
"""Coroutine which look to "{...}" which are RdLeaf
|
||||||
|
|
||||||
:example:
|
:example:
|
||||||
>>> from ...core.str2 import list_sink
|
>>> from ...core.str2 import list_sink
|
||||||
@ -53,9 +53,7 @@ def look_for_rdleaf(target):
|
|||||||
|
|
||||||
|
|
||||||
class RdLeaf:
|
class RdLeaf:
|
||||||
""" Random leaf
|
"""Random leaf"""
|
||||||
|
|
||||||
"""
|
|
||||||
|
|
||||||
def __init__(self, name):
|
def __init__(self, name):
|
||||||
self._name = name
|
self._name = name
|
||||||
|
@ -4,8 +4,9 @@ from .grammar import extract_letters, eval_words
|
|||||||
from .generate import random_generator
|
from .generate import random_generator
|
||||||
from .str2 import rdstr2
|
from .str2 import rdstr2
|
||||||
|
|
||||||
|
|
||||||
class RandomTree(MutableTree):
|
class RandomTree(MutableTree):
|
||||||
""" MutableTree that accept {a} syntax for random generation
|
"""MutableTree that accept {a} syntax for random generation
|
||||||
|
|
||||||
:example:
|
:example:
|
||||||
>>> t = RandomTree()
|
>>> t = RandomTree()
|
||||||
@ -15,7 +16,7 @@ class RandomTree(MutableTree):
|
|||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def from_str(cls, expression):
|
def from_str(cls, expression):
|
||||||
""" Initiate a random tree from a string that need to be parsed
|
"""Initiate a random tree from a string that need to be parsed
|
||||||
|
|
||||||
:exemple:
|
:exemple:
|
||||||
>>> t = RandomTree.from_str("{b}*x+{c}")
|
>>> t = RandomTree.from_str("{b}*x+{c}")
|
||||||
@ -40,7 +41,7 @@ class RandomTree(MutableTree):
|
|||||||
|
|
||||||
@property
|
@property
|
||||||
def random_leaves(self) -> list[str]:
|
def random_leaves(self) -> list[str]:
|
||||||
""" Get list of random leaves
|
"""Get list of random leaves
|
||||||
|
|
||||||
:example:
|
:example:
|
||||||
>>> from .leaf import RdLeaf
|
>>> from .leaf import RdLeaf
|
||||||
@ -61,10 +62,9 @@ class RandomTree(MutableTree):
|
|||||||
rd_leafs.append(leaf.name)
|
rd_leafs.append(leaf.name)
|
||||||
return rd_leafs
|
return rd_leafs
|
||||||
|
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def random_value(self) -> set[str]:
|
def random_value(self) -> set[str]:
|
||||||
""" Get set of random values to generate
|
"""Get set of random values to generate
|
||||||
|
|
||||||
:example:
|
:example:
|
||||||
>>> from .leaf import RdLeaf
|
>>> from .leaf import RdLeaf
|
||||||
@ -77,9 +77,8 @@ class RandomTree(MutableTree):
|
|||||||
"""
|
"""
|
||||||
return extract_letters(self.random_leaves)
|
return extract_letters(self.random_leaves)
|
||||||
|
|
||||||
|
def eval_random_leaves(self, leaves_value: dict[str, int]):
|
||||||
def eval_random_leaves(self, leaves_value:dict[str, int]):
|
"""Given random leaves value get the tree
|
||||||
""" Given random leaves value get the tree
|
|
||||||
|
|
||||||
:example:
|
:example:
|
||||||
>>> from .leaf import RdLeaf
|
>>> from .leaf import RdLeaf
|
||||||
@ -93,21 +92,27 @@ class RandomTree(MutableTree):
|
|||||||
> 2
|
> 2
|
||||||
> 6
|
> 6
|
||||||
"""
|
"""
|
||||||
|
|
||||||
def replace(leaf):
|
def replace(leaf):
|
||||||
try:
|
try:
|
||||||
return leaf.replace(leaves_value)
|
return leaf.replace(leaves_value)
|
||||||
except AttributeError:
|
except AttributeError:
|
||||||
return leaf
|
return leaf
|
||||||
|
|
||||||
return self.map_on_leaf(replace).map_on_leaf(moify)
|
return self.map_on_leaf(replace).map_on_leaf(moify)
|
||||||
|
|
||||||
def generate(self, conditions:list[str]=[], global_config:dict={} , configs:dict={}) -> Tree:
|
def generate(
|
||||||
""" Generate a random version of self
|
self, conditions: list[str] = [], global_config: dict = {}, configs: dict = {}
|
||||||
|
) -> Tree:
|
||||||
|
"""Generate a random version of self
|
||||||
|
|
||||||
:param conditions: list of conditions
|
:param conditions: list of conditions
|
||||||
:param config: global configuration for generated values
|
:param config: global configuration for generated values
|
||||||
:param configs: specific configuration for each generated values
|
:param configs: specific configuration for each generated values
|
||||||
|
|
||||||
"""
|
"""
|
||||||
generated_values = random_generator(self.random_value, conditions, global_config, configs)
|
generated_values = random_generator(
|
||||||
|
self.random_value, conditions, global_config, configs
|
||||||
|
)
|
||||||
leaves = eval_words(self.random_leaves, generated_values)
|
leaves = eval_words(self.random_leaves, generated_values)
|
||||||
return self.eval_random_leaves(leaves)
|
return self.eval_random_leaves(leaves)
|
||||||
|
@ -1,13 +1,19 @@
|
|||||||
|
|
||||||
from ...core.operator import is_operator
|
from ...core.operator import is_operator
|
||||||
from functools import partial
|
from functools import partial
|
||||||
from ...core.str2 import concurent_broadcast, lookforNumbers, pparser, missing_times, lookfor
|
from ...core.str2 import (
|
||||||
|
concurent_broadcast,
|
||||||
|
lookforNumbers,
|
||||||
|
pparser,
|
||||||
|
missing_times,
|
||||||
|
lookfor,
|
||||||
|
)
|
||||||
from ...core.coroutine import STOOOP
|
from ...core.coroutine import STOOOP
|
||||||
from ...core.MO import moify_cor
|
from ...core.MO import moify_cor
|
||||||
from .leaf import look_for_rdleaf
|
from .leaf import look_for_rdleaf
|
||||||
|
|
||||||
|
|
||||||
def rdstr2(sink):
|
def rdstr2(sink):
|
||||||
""" Return a pipeline which parse random expression and with sink as endpoint
|
"""Return a pipeline which parse random expression and with sink as endpoint
|
||||||
|
|
||||||
:example:
|
:example:
|
||||||
>>> from ...core.str2 import list_sink
|
>>> from ...core.str2 import list_sink
|
||||||
@ -22,8 +28,7 @@ def rdstr2(sink):
|
|||||||
|
|
||||||
def pipeline(expression):
|
def pipeline(expression):
|
||||||
str2_corout = look_for_rdleaf(
|
str2_corout = look_for_rdleaf(
|
||||||
lookforNumbers(operator_corout(
|
lookforNumbers(operator_corout(missing_times(moify_cor(pparser(sink)))))
|
||||||
missing_times(moify_cor(pparser(sink)))))
|
|
||||||
)
|
)
|
||||||
|
|
||||||
for i in expression.replace(" ", ""):
|
for i in expression.replace(" ", ""):
|
||||||
@ -33,4 +38,3 @@ def rdstr2(sink):
|
|||||||
return a
|
return a
|
||||||
|
|
||||||
return pipeline
|
return pipeline
|
||||||
|
|
||||||
|
@ -4,10 +4,16 @@ from .core.random_tree import RandomTree
|
|||||||
DEFAUTL_CONFIG = {
|
DEFAUTL_CONFIG = {
|
||||||
"rejected": [0, 1],
|
"rejected": [0, 1],
|
||||||
"min_max": (-10, 10),
|
"min_max": (-10, 10),
|
||||||
}
|
}
|
||||||
|
|
||||||
def expression_generator(template:str, conditions:list[str]=[], global_config:dict={}, configs:dict={}):
|
|
||||||
""" Generate a random expression
|
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 template: the template of the expression
|
||||||
:param conditions: conditions on randomly generate variable
|
:param conditions: conditions on randomly generate variable
|
||||||
@ -31,5 +37,7 @@ def expression_generator(template:str, conditions:list[str]=[], global_config:di
|
|||||||
|
|
||||||
"""
|
"""
|
||||||
rd_tree = RandomTree.from_str(template)
|
rd_tree = RandomTree.from_str(template)
|
||||||
generated_tree = rd_tree.generate(conditions, dict(DEFAUTL_CONFIG, **global_config), configs)
|
generated_tree = rd_tree.generate(
|
||||||
|
conditions, dict(DEFAUTL_CONFIG, **global_config), configs
|
||||||
|
)
|
||||||
return Expression._post_processing(generated_tree)
|
return Expression._post_processing(generated_tree)
|
||||||
|
@ -15,10 +15,16 @@ from .core.grammar import extract_letters, eval_words
|
|||||||
DEFAUTL_CONFIG = {
|
DEFAUTL_CONFIG = {
|
||||||
"rejected": [0],
|
"rejected": [0],
|
||||||
"min_max": (-10, 10),
|
"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
|
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 rd_variables: list of random variables to generate (can be computed value - "a*b")
|
||||||
:param conditions: condition over variables
|
:param conditions: condition over variables
|
||||||
@ -51,6 +57,8 @@ def list_generator(var_list:list[str], conditions:list[str]=[], global_config:di
|
|||||||
True
|
True
|
||||||
"""
|
"""
|
||||||
rv = extract_letters(var_list)
|
rv = extract_letters(var_list)
|
||||||
rv_gen = random_generator(rv, conditions, dict(DEFAUTL_CONFIG, **global_config), configs)
|
rv_gen = random_generator(
|
||||||
|
rv, conditions, dict(DEFAUTL_CONFIG, **global_config), configs
|
||||||
|
)
|
||||||
generated = eval_words(var_list, rv_gen)
|
generated = eval_words(var_list, rv_gen)
|
||||||
return [generated[v] for v in var_list]
|
return [generated[v] for v in var_list]
|
||||||
|
Loading…
Reference in New Issue
Block a user