Cleans up imports. Switches from junit.framework.Assert.assertEquals (which is deprecated)
to org.junit.Assert.assertEquals.