diff --git a/test/calculus/random/test_list_generator.py b/test/calculus/random/test_list_generator.py index f4bf747..2960fcd 100644 --- a/test/calculus/random/test_list_generator.py +++ b/test/calculus/random/test_list_generator.py @@ -32,5 +32,24 @@ def test_generate_list_calculus_math(): def test_generate_list_conditions(): - random_list = mapytex.random.list(["a", "b"], ) + a, b = mapytex.random.list(["a", "b"], conditions=["a + b == 10"]) + assert a + b == 10 + a, b = mapytex.random.list(["a", "b"], conditions=["a * b > 0", "a + b == 10"]) + assert a + b == 10 + assert a * b > 0 + +def test_generate_list_conditions_math(): + import math + a, b = mapytex.random.list(["a", "b"], conditions=["gcd(a, b) == 3"]) + assert math.gcd(a, b) == 3 + +def test_generate_list_global_config(): + global_config = {"rejected": [0, 1, 2, 3]} + a, = mapytex.random.list(["a"], global_config=global_config) + assert a not in global_config["rejected"] + + global_config = {"min_max": (20, 30)} + a, = mapytex.random.list(["a"], global_config=global_config) + assert a >= 20 + assert a <= 30