56 lines
1.9 KiB
Python
56 lines
1.9 KiB
Python
import mapytex
|
|
|
|
def test_generate_list():
|
|
random_list = mapytex.random.list(["a", "b"])
|
|
assert len(random_list) == 2
|
|
random_list = mapytex.random.list(["a", "b", "c"])
|
|
assert len(random_list) == 3
|
|
random_list = mapytex.random.list(["a", "b", "a", "b"])
|
|
assert random_list[0] == random_list[2]
|
|
assert random_list[1] == random_list[3]
|
|
|
|
def test_generate_list_calculus():
|
|
random_list = mapytex.random.list(["a", "b", "a+b"])
|
|
assert random_list[0] + random_list[1] == random_list[2]
|
|
|
|
random_list = mapytex.random.list(["a", "b", "a-b"])
|
|
assert random_list[0] - random_list[1] == random_list[2]
|
|
|
|
random_list = mapytex.random.list(["a", "b", "a*b"])
|
|
assert random_list[0] * random_list[1] == random_list[2]
|
|
|
|
random_list = mapytex.random.list(["a", "b", "a/b"])
|
|
assert random_list[0] / random_list[1] == random_list[2]
|
|
|
|
def test_generate_list_calculus_math():
|
|
import math
|
|
a, b, gcd = mapytex.random.list(["a", "b", "gcd(a, b)"])
|
|
assert math.gcd(a, b) == gcd
|
|
a, b, exp, log = mapytex.random.list(["a", "b", "exp(a)", "log(b)"])
|
|
assert math.exp(a) == exp
|
|
assert math.log(b) == log
|
|
|
|
|
|
def test_generate_list_conditions():
|
|
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
|
|
|