cts_common_test.py 390 B

1234567891011121314
  1. import pytest
  2. from typing import List
  3. from dw_base.spark.udf.customs.cts_common import explode_str_to_arr
  4. @pytest.mark.parametrize("text, expected", [
  5. ('abc', ['abc']),
  6. ('abcdefghi',['abcdefghi', 'abcdefgh']),
  7. ('xxxxaaaa',['xxxxaaaa']),
  8. (None,[]),
  9. ('',[''])
  10. ])
  11. def test_explode_str_to_arr(text:str, expected: List[str]):
  12. assert explode_str_to_arr(text) == expected