Replace junit.framework.Assert with org.junit.Assert