Fix async prediction tests
These tests relied on soon-to-be-removed behaviour of cog where we
guaranteed that we'd return an "initial" state for the prediction when
"Prefer: respond-async" was specified.
We will shortly allow fast-completing predictions to respond with the
current state of the prediction, so the tests need to be a little more
flexible with what they assert.