Use testify/assert for test cases.

[MarkGibbons]

We have a mix of testing methods. I think testify/assert will work pretty well.