No need to write the same assertion in each case (except the default
one). Instead we can assert after the switch statement.