Updated the unit test cases to fix Travis fail

Signed-off-by: Vivek Rai <vivekraiiitkgp@gmail.com>