Update one test to use unittest.

I'm likely to do this for all tests, but this one's a start.