Add tests for wrong input shapes