modified setup method

Updated expected result for unittest setup method