Add expectations tests