Added test for one_shot option