Add a component test for batch_ax_function_py
The code had a problem that got fixed in !2838 (merged) . It would be good to add a unit test for it.
The code had a problem that got fixed in !2838 (merged) . It would be good to add a unit test for it.