def test_divide_purely_imaginary_numbers(self):
input_number = ComplexNumber(0, 1)
expected = ComplexNumber(0.5, 0)
divider = ComplexNumber(0, 2)
self.assertEqual(input_number.div(divider).real, expected.real)
self.assertEqual(
input_number.div(divider).imaginary, expected.imaginary)
def test_divide_numbers_with_real_and_imaginary_part(self):
input_number = ComplexNumber(1, 2)
expected = ComplexNumber(0.44, 0.08)
divider = ComplexNumber(3, 4)
self.assertEqual(input_number.div(divider).real, expected.real)
self.assertEqual(
input_number.div(divider).imaginary, expected.imaginary)
def test_divide_purely_real_numbers(self):
input_number = ComplexNumber(1.0, 0.0)
expected = ComplexNumber(0.5, 0.0)
divider = ComplexNumber(2.0, 0.0)
self.assertEqual(expected.real, input_number.div(divider).real)
self.assertEqual(expected.imaginary,
input_number.div(divider).imaginary)
