Fixed magic numbers

We had some magic numbers in the tests, so I've removed them and made them constants.