For many repos we have replaced pingcap/check with testify. We could do the same here. Currently this repo uses both testify and pingcap/check.

This makes this more consistent with what we do for other repositories.