improving some tests

Signed-off-by: jayrambhia <jayrambhia777@gmail.com>