[SPARK-40393][PS][TESTS] Refactor expanding and rolling test for function with input

[Yikun]

What changes were proposed in this pull request?

Refactor expanding and rolling test for function with input

Why are the changes needed?

Refactor expanding and rolling test for function with input:

# Before
self._test_groupby_rolling_func("count")

# After
# str can be accept
self._test_groupby_rolling_func("count")
# Can also accept lambda to support more func style
self._test_groupby_expanding_func(
    lambda x: x.quantile(0.5), lambda x: x.quantile(0.5, "lower")
)

Does this PR introduce any user-facing change?

No

How was this patch tested?

• CI passed
• cherry-pick: e22ee1c and test manually.

[Yikun]

cc @zhengruifeng @HyukjinKwon

[itholic]

Looks good otherwise

[itholic]

Sorry, I got a bit confusion for this part.

Could you tell one example that considers this case?

[HyukjinKwon]

I think it's that it uses the same function if pd_func is omitted.

[Yikun]

Yes, just like: self._test_groupby_rolling_func("count").

We add ps_func and pd_func separately for some case like: self._test_groupby_expanding_func(lambda x:x.quantile(0.5), lambda x: x.quantile(0.5, "lower")), pd_func and ps_func is different.

[HyukjinKwon]

Merged to master.