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)
