def test_add_purely_imaginary_numbers(self): first_input = ComplexNumber(0, 1) second_input = ComplexNumber(0, 2) self.assertEqual(first_input.add(second_input).real, 0) self.assertEqual(first_input.add(second_input).imaginary, 3)
def test_add_numbers_with_real_and_imaginary_part(self): first_input = ComplexNumber(1, 2) second_input = ComplexNumber(3, 4) self.assertEqual(first_input.add(second_input).real, 4) self.assertEqual(first_input.add(second_input).imaginary, 6)
def test_add_purely_real_numbers(self): first_input = ComplexNumber(1, 0) second_input = ComplexNumber(2, 0) self.assertEqual(first_input.add(second_input).real, 3) self.assertEqual(first_input.add(second_input).imaginary, 0)