import junit.framework.*;

// A demonstration of JUnit testing
public class ComplexNumberTest extends TestCase
{
   ComplexNumber cn1, cn2;
   private double testR = 6;

   public ComplexNumberTest (String name)
   {
      super (name);
   }

    // Create variables to used in each test case
   protected void setUp()
   {
      cn1 = new ComplexNumber(6, 8);
      cn2 = new ComplexNumber(1, 3);
   }

   /** @author Wilson Lau */
   public void testgetReal()
   {
      cn1.toString();
      assertEquals(cn1.getReal(),testR);
   }

   /* @author Jason Boyle */
   public void testEquals()
   {
      ComplexNumber cnA = new ComplexNumber(4, 6);
      ComplexNumber cnB = new ComplexNumber(4, 6);
      assertEquals(cnA, cnB);
   }
   /**
    * Tests the 'subtract' methods in ComplexNumber.java
    *
    *@author Mike Murray
    */
   public void testSubtraction()
   {
      /* Test case of subtracting a complex number from instance */
      assertEquals(cn1.subtract(new ComplexNumber(5, 5)), cn2);
      /* Test case of result being 0 from subtraction */
      assertEquals(cn1.subtract(cn1), new ComplexNumber(0, 0));
      /* Test subtraction of two declared complex numbers */
      assertEquals(ComplexNumber.subtract(cn2, cn1),
                   new ComplexNumber(-5, -5));
   }
   /* @author Nick Patel */
   public void testDivision()
   {
       ComplexNumber cnDiv1= new ComplexNumber(2,-3);
       ComplexNumber cnDiv2= new ComplexNumber(3,-4);
       ComplexNumber cnTotal = new ComplexNumber(.72 , -.04);
       cnDiv1=ComplexNumber.divide(cnDiv1,cnDiv2);
       assertEquals(cnDiv1, cnTotal);
   }

   /* @author Nestor Reyes */

   public void testMagnitude()
   {
       ComplexNumber cnmag1 = new ComplexNumber(2, 4);
       ComplexNumber cnmag2 = new ComplexNumber(-52, 90);

       assertEquals(cnmag1.magnitude(), Math.sqrt(20), 0);
       assertEquals(cnmag2.magnitude(), Math.sqrt(52*52 + 90*90), 0);
   }
}