import pytest from dw_base.spark.udf.enterprise.ent_logistics_label import is_logistic_match @pytest.mark.parametrize("name, expected",[ ('ALINE', False)]) def test_is_logistic_match(name, expected): result = is_logistic_match(name) assert result == expected if __name__ == '__main__': pytest.main()