Modified structure of test_methods and adding test case in new_returns_flow_spec