Refactored unit test imports from deprecated junit.framework.Assert to org.junit.Assert