danvics/ytdl-sub
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

bool funcs

Browse source
This commit is contained in:
Jesse Bannon 2023-11-30 07:30:19 -08:00
parent f76e421b2d
commit c0f6eca3c6
2 changed files with 201 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
src/ytdl_sub/script/functions/boolean_functions.py
View file

@ -59,26 +59,27 @@ class BooleanFunctions:
return Boolean(left.value >= right.value)
@staticmethod
def and_(left: Boolean, right: Boolean) -> Boolean:
def and_(*values: AnyArgument) -> Boolean:
"""
``and`` operator. Returns True if left and right are both True. False otherwise.
``and`` operator. Returns True if all values evaluate to True. False otherwise.
"""
return Boolean(left.value and right.value)
return Boolean(all(bool(val.value) for val in values))
@staticmethod
def or_(left: Boolean, right: Boolean) -> Boolean:
def or_(*values: AnyArgument) -> Boolean:
"""
``or`` operator. Returns True if either left or right are True. False otherwise.
``or`` operator. Returns True if any value evaluates to True. False otherwise.
"""
return Boolean(left.value or right.value)
return Boolean(any(bool(val.value) for val in values))
@staticmethod
def xor(left: Boolean, right: Boolean) -> Boolean:
def xor(*values: AnyArgument) -> Boolean:
"""
``^`` operator. Returns True if either left or right are True and the other is False.
Returns False otherwise.
``^`` operator. Returns True if exactly one value is set to True. False otherwise.
"""
return Boolean(left.value ^ right.value)
bit_array = [bool(val.value) for val in values]
return Boolean(sum(bit_array) == 1)
@staticmethod
def not_(value: Boolean) -> Boolean:

190
tests/unit/script/functions/test_boolean_functions.py Normal file
View file

@ -0,0 +1,190 @@
import pytest
from ytdl_sub.script.script import Script
class TestBooleanFunctions:
@pytest.mark.parametrize(
"lhs, rhs, expected_output",
[
("'abc'", "'abc'", True),
("123", "123", True),
("3.14", "3.14", True),
("True", "True", True),
("False", "False", True),
("[1, 2, 3]", "[1, 2, 3]", True),
("{'key': 'value'}", "{'key': 'value'}", True),
("{'key': 'value'}", 5, False),
("{'key': 'value'}", "[1, 2, 3]", False),
],
)
@pytest.mark.parametrize("is_ne", [True, False])
def test_eq_ne(self, lhs: str, rhs: str, expected_output: bool, is_ne: bool):
op = "ne" if is_ne else "eq"
output = (
Script(
{
"output": f"{{%{op}({lhs}, {rhs})}}",
}
)
.resolve(update=True)
.get("output")
.native
)
if is_ne:
assert output != expected_output
else:
assert output == expected_output
@pytest.mark.parametrize(
"lhs, rhs, expected_output",
[
("'abc'", "'abc'", True),
("123", "123", True),
("3.14", "3.14", True),
("3.14", "4.0", True),
],
)
@pytest.mark.parametrize("is_gt", [True, False])
def test_lte_gt(self, lhs: str, rhs: str, expected_output: bool, is_gt: bool):
op = "gt" if is_gt else "lte"
output = (
Script(
{
"output": f"{{%{op}({lhs}, {rhs})}}",
}
)
.resolve(update=True)
.get("output")
.native
)
if is_gt:
assert output != expected_output
else:
assert output == expected_output
@pytest.mark.parametrize(
"lhs, rhs, expected_output",
[
("'abc'", "'abc'", True),
("123", "123", True),
("3.14", "3.14", True),
("5.32", "4", True),
("3.14", "4.0", False),
],
)
@pytest.mark.parametrize("is_lt", [True, False])
def test_gte_lt(self, lhs: str, rhs: str, expected_output: bool, is_lt: bool):
op = "lt" if is_lt else "gte"
output = (
Script(
{
"output": f"{{%{op}({lhs}, {rhs})}}",
}
)
.resolve(update=True)
.get("output")
.native
)
if is_lt:
assert output != expected_output
else:
assert output == expected_output
@pytest.mark.parametrize(
"values, expected_output",
[
("True", True),
("True, True", True),
("True, True, True", True),
("False", False),
("False, True", False),
("True, False, True", False),
],
)
def test_and(self, values: str, expected_output: bool):
output = (
Script(
{
"output": f"{{%and({values})}}",
}
)
.resolve(update=True)
.get("output")
.native
)
assert output == expected_output
@pytest.mark.parametrize(
"values, expected_output",
[
("True", True),
("True, True", True),
("True, True, True", True),
("False", False),
("False, True", True),
("True, False, True", True),
("False, False, False", False),
],
)
def test_or(self, values: str, expected_output: bool):
output = (
Script(
{
"output": f"{{%or({values})}}",
}
)
.resolve(update=True)
.get("output")
.native
)
assert output == expected_output
@pytest.mark.parametrize(
"values, expected_output",
[
("True", True),
("True, True", False),
("True, True, True", False),
("False", False),
("False, True", True),
("True, False, True", False),
("False, False, False", False),
("False, True, False", True),
],
)
def test_xor(self, values: str, expected_output: bool):
output = (
Script(
{
"output": f"{{%xor({values})}}",
}
)
.resolve(update=True)
.get("output")
.native
)
assert output == expected_output
@pytest.mark.parametrize(
"value, expected_output",
[
("True", False),
("False", True),
],
)
def test_not(self, value: str, expected_output: bool):
output = (
Script(
{
"output": f"{{%not({value})}}",
}
)
.resolve(update=True)
.get("output")
.native
)
assert output == expected_output