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

condtional functions

Browse source
This commit is contained in:
Jesse Bannon 2023-11-30 07:44:33 -08:00
parent 80c32335fb
commit d35c8e9965
2 changed files with 35 additions and 35 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
tests/unit/script/functions/test_conditional_functions.py Normal file
View file

@ -0,0 +1,35 @@
import pytest
from ytdl_sub.script.script import Script
class TestConditionalFunction:
@pytest.mark.parametrize(
"function_str, expected_output",
[
("{%if(True, True, False)}", True),
("{%if(False, True, False)}", False),
],
)
def test_if_function(self, function_str: str, expected_output: bool):
output = Script({"output": function_str}).resolve(update=True).get("output").native
assert output == expected_output
def test_nested_if_function(self):
function_str = """{
%if(
True,
%if(
True,
%if(
True,
"winner",
True
),
True
),
True
)
}"""
output = Script({"output": function_str}).resolve(update=True).get("output").native
assert output == "winner"

35
tests/unit/script/types/test_function.py
View file

@ -4,13 +4,10 @@ import pytest
from ytdl_sub.script.parser import FUNCTION_INVALID_CHAR from ytdl_sub.script.parser import FUNCTION_INVALID_CHAR
from ytdl_sub.script.script import Script from ytdl_sub.script.script import Script
from ytdl_sub.script.types.resolvable import Boolean
from ytdl_sub.script.types.resolvable import String
from ytdl_sub.script.utils.exceptions import FunctionDoesNotExist from ytdl_sub.script.utils.exceptions import FunctionDoesNotExist
from ytdl_sub.script.utils.exceptions import FunctionRuntimeException from ytdl_sub.script.utils.exceptions import FunctionRuntimeException
from ytdl_sub.script.utils.exceptions import IncompatibleFunctionArguments from ytdl_sub.script.utils.exceptions import IncompatibleFunctionArguments
from ytdl_sub.script.utils.exceptions import InvalidSyntaxException from ytdl_sub.script.utils.exceptions import InvalidSyntaxException
from ytdl_sub.script.utils.exceptions import UserThrownRuntimeError
def _incompatible_arguments_match(expected: str, recieved: str) -> str: def _incompatible_arguments_match(expected: str, recieved: str) -> str:
@ -18,38 +15,6 @@ def _incompatible_arguments_match(expected: str, recieved: str) -> str:
class TestFunction: class TestFunction:
@pytest.mark.parametrize(
"function_str, expected_output",
[
("{%if(True, True, False)}", True),
("{%if(False, True, False)}", False),
],
)
def test_if_function(self, function_str: str, expected_output: bool):
assert Script({"func": function_str}).resolve() == {
"func": Boolean(expected_output),
}
def test_nested_if_function(self):
function_str = """{
%if(
True,
%if(
True,
%if(
True,
"winner",
True
),
True
),
True
)
}"""
assert Script({"func": function_str}).resolve() == {
"func": String("winner"),
}
def test_nested_if_function_incompatible(self): def test_nested_if_function_incompatible(self):
function_str = """{ function_str = """{
%map_get( %map_get(