f04087b389
Migrated all interfaces to get and return strings. Added tests and evals for all functions (except the random generation) Math functions are now organized into different math categories --------- Co-authored-by: Nate Barbettini <nate@arcade-ai.com>
142 lines
3.6 KiB
Python
142 lines
3.6 KiB
Python
import pytest
|
|
from arcade.sdk.errors import ToolExecutionError
|
|
|
|
from arcade_math.tools.arithmetic import (
|
|
add,
|
|
divide,
|
|
mod,
|
|
multiply,
|
|
subtract,
|
|
sum_list,
|
|
sum_range,
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"a, b, expected",
|
|
[
|
|
("1", "2", "3"),
|
|
("-1", "1", "0"),
|
|
("0.5", "10.9", "11.4"),
|
|
# Big ints
|
|
("12345678901234567890", "9876543210987654321", "22222222112222222211"),
|
|
# Big floats
|
|
(
|
|
"12345678901234567890.120",
|
|
"9876543210987654321.987",
|
|
"22222222112222222212.107",
|
|
),
|
|
],
|
|
)
|
|
def test_add(a, b, expected):
|
|
assert add(a, b) == expected
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"a, b, expected",
|
|
[
|
|
("1", "2", "-1"),
|
|
("-1", "1", "-2"),
|
|
("0.5", "10.9", "-10.4"),
|
|
# Big ints
|
|
("12345678901234567890", "12323456679012345668", "22222222222222222"),
|
|
# Big floats
|
|
(
|
|
"12345678901234567890.120",
|
|
"12343557689113355768.9079",
|
|
"2121212121212121.2121",
|
|
),
|
|
],
|
|
)
|
|
def test_subtract(a, b, expected):
|
|
assert subtract(a, b) == expected
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"a, b, expected",
|
|
[
|
|
("-1", "2", "-2"),
|
|
("-10", "0", "-0"),
|
|
("0.5", "10.9", "5.45"),
|
|
# Big ints
|
|
(
|
|
"12345678901234567890",
|
|
"18000000162000001474380013420000",
|
|
"2.222222222222222222222222223E+50",
|
|
),
|
|
# Big floats
|
|
(
|
|
"12345678901234567890.120",
|
|
"12345678901234567890.120",
|
|
"1.524157875323883675048681628E+38",
|
|
),
|
|
],
|
|
)
|
|
def test_multiply(a, b, expected):
|
|
assert multiply(a, b) == expected
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"a, b, expected",
|
|
[
|
|
("-1", "2", "-0.5"),
|
|
("-10", "1", "-10"),
|
|
("0.5", "10.9", "0.04587155963302752293577981651"),
|
|
# Big ints
|
|
("152407406035740740602050", "12345678901234567890", "12345"),
|
|
# Big floats
|
|
(
|
|
"152407406035740740603531.400",
|
|
"12345678901234567890.120",
|
|
"12345",
|
|
),
|
|
],
|
|
)
|
|
def test_divide(a, b, expected):
|
|
assert divide(a, b) == expected
|
|
|
|
|
|
def text_zero_division():
|
|
with pytest.raises(ToolExecutionError):
|
|
divide("1", "0")
|
|
with pytest.raises(ToolExecutionError):
|
|
divide("1", "0.0")
|
|
with pytest.raises(ToolExecutionError):
|
|
divide("1", "0.000000")
|
|
|
|
|
|
def test_sum_list():
|
|
assert sum_list(["1", "2", "3", "4", "5", "6"]) == "21"
|
|
assert sum_list([]) == "0"
|
|
assert sum_list(["-1", "-2", "-3", "-4", "-5", "-6"]) == "-21"
|
|
assert sum_list(["0.1", "0.2", "0.3", "0.3", "0.5", "0.7"]) == "2.1"
|
|
|
|
|
|
def test_sum_range():
|
|
assert sum_range("8", "2") == "0"
|
|
assert sum_range("-8", "2") == "-33"
|
|
assert sum_range("8", "-2") == "0"
|
|
assert sum_range("2", "3") == "5"
|
|
assert sum_range("0", "10") == "55"
|
|
with pytest.raises(ToolExecutionError):
|
|
sum_range("2", "0.5")
|
|
with pytest.raises(ToolExecutionError):
|
|
sum_range("-1", "0.5")
|
|
with pytest.raises(ToolExecutionError):
|
|
sum_range("2.", "0.5")
|
|
with pytest.raises(ToolExecutionError):
|
|
sum_range("-1", "0.5")
|
|
|
|
|
|
def test_mod():
|
|
assert mod("-1", "0.5") == "-0.0"
|
|
assert mod("-8", "2") == "-0"
|
|
assert mod("0", "10") == "0"
|
|
assert mod("2", "0.5") == "0.0"
|
|
assert mod("2", "3") == "2"
|
|
assert mod("2.", "-0.5") == "0.0"
|
|
assert mod("2.1234", "0.6") == "0.3234"
|
|
assert mod("2.1234", "1") == "0.1234"
|
|
assert mod("2.1234", "3") == "2.1234"
|
|
assert mod("8", "-2") == "0"
|
|
assert mod("8", "2") == "0"
|