use Expected schema
add test cases