Fix: format with black

This commit is contained in:
Bertrand Benjamin 2021-10-09 06:30:38 +02:00
parent a267acd7b3
commit aad2395a3a
8 changed files with 140 additions and 119 deletions

View File

@ -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

View File

@ -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}

View File

@ -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))

View File

@ -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

View File

@ -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)

View File

@ -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

View File

@ -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)

View File

@ -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]