review changes for test cases

Signed-off-by: Nick Desai <ndesai36@bloomberg.net>