Use JUnit 4 assertions instead of JUnit 3.

Should be painless.