import pytest
from typing import List

from dw_base.spark.udf.customs.cts_common import explode_str_to_arr

@pytest.mark.parametrize("text, expected", [
    ('abc', ['abc']),
    ('abcdefghi',['abcdefghi', 'abcdefgh']),
    ('xxxxaaaa',['xxxxaaaa']),
    (None,[]),
    ('',[''])
])
def test_explode_str_to_arr(text:str, expected: List[str]):
    assert explode_str_to_arr(text) == expected