add test for simple cases