add additional test cases

Signed-off-by: Ethan Vizitei <ethan.vizitei@gmail.com>