Replaced assert_raises with assert_raise_message as the point is to make sure the exception message is clear.