Replace assumptions with guarantees

This makes the tests faster and avoids hypothesis complaints about assumptions
that fail too often.