import static org.junit.Assert.*; import org.junit.After; import org.junit.Before; import org.junit.Test; public class SquaresTest { Squares square=new Squares(); @Test public void testSquare() { int expected=1; int actual=square.display(1); assertEquals(actual,expected); } }