Move tests to JUnit

Related to #13.