Move tests to JUnit.
Closes #9.