revise generator test cases