be794198e1
* Corrected small typos around resistor bands. * Due to failing CI, alterations to the test generator script were needed. The generated vs submitted diff now skips the first three lines of the file so that the generation date is not picked up and flagged as needing regeneration. Sadly, a workaround was also needed to prevent Python difflib from noting the difference anyways and producing an empty "false positive" diff. All templates and test files also needed to be altered to ensure that the first three lines of every test file will always be the autogeneration comment and date. Hopefully, this will now stop the CI failures without creating any subtle additional bugs. * Touch up to bowling template. Added back the error raising utility. * Touch up to two-bucket template to add back in error raising utility. [no important files changed]
193 lines
6.6 KiB
Python
193 lines
6.6 KiB
Python
# These tests are auto-generated with test data from:
|
|
# https://github.com/exercism/problem-specifications/tree/main/exercises/complex-numbers/canonical-data.json
|
|
# File last updated on 2023-07-19
|
|
|
|
import math
|
|
import unittest
|
|
|
|
from complex_numbers import (
|
|
ComplexNumber,
|
|
)
|
|
|
|
|
|
class ComplexNumbersTest(unittest.TestCase):
|
|
|
|
# Real part
|
|
|
|
def test_real_part_of_a_purely_real_number(self):
|
|
self.assertEqual(ComplexNumber(1, 0).real, 1)
|
|
|
|
def test_real_part_of_a_purely_imaginary_number(self):
|
|
self.assertEqual(ComplexNumber(0, 1).real, 0)
|
|
|
|
def test_real_part_of_a_number_with_real_and_imaginary_part(self):
|
|
self.assertEqual(ComplexNumber(1, 2).real, 1)
|
|
|
|
# Imaginary part
|
|
|
|
def test_imaginary_part_of_a_purely_real_number(self):
|
|
self.assertEqual(ComplexNumber(1, 0).imaginary, 0)
|
|
|
|
def test_imaginary_part_of_a_purely_imaginary_number(self):
|
|
self.assertEqual(ComplexNumber(0, 1).imaginary, 1)
|
|
|
|
def test_imaginary_part_of_a_number_with_real_and_imaginary_part(self):
|
|
self.assertEqual(ComplexNumber(1, 2).imaginary, 2)
|
|
|
|
def test_imaginary_unit(self):
|
|
self.assertEqual(
|
|
ComplexNumber(0, 1) * ComplexNumber(0, 1), ComplexNumber(-1, 0)
|
|
)
|
|
|
|
# Arithmetic
|
|
|
|
# Addition
|
|
|
|
def test_add_purely_real_numbers(self):
|
|
self.assertEqual(ComplexNumber(1, 0) + ComplexNumber(2, 0), ComplexNumber(3, 0))
|
|
|
|
def test_add_purely_imaginary_numbers(self):
|
|
self.assertEqual(ComplexNumber(0, 1) + ComplexNumber(0, 2), ComplexNumber(0, 3))
|
|
|
|
def test_add_numbers_with_real_and_imaginary_part(self):
|
|
self.assertEqual(ComplexNumber(1, 2) + ComplexNumber(3, 4), ComplexNumber(4, 6))
|
|
|
|
# Subtraction
|
|
|
|
def test_subtract_purely_real_numbers(self):
|
|
self.assertEqual(
|
|
ComplexNumber(1, 0) - ComplexNumber(2, 0), ComplexNumber(-1, 0)
|
|
)
|
|
|
|
def test_subtract_purely_imaginary_numbers(self):
|
|
self.assertEqual(
|
|
ComplexNumber(0, 1) - ComplexNumber(0, 2), ComplexNumber(0, -1)
|
|
)
|
|
|
|
def test_subtract_numbers_with_real_and_imaginary_part(self):
|
|
self.assertEqual(
|
|
ComplexNumber(1, 2) - ComplexNumber(3, 4), ComplexNumber(-2, -2)
|
|
)
|
|
|
|
# Multiplication
|
|
|
|
def test_multiply_purely_real_numbers(self):
|
|
self.assertEqual(ComplexNumber(1, 0) * ComplexNumber(2, 0), ComplexNumber(2, 0))
|
|
|
|
def test_multiply_purely_imaginary_numbers(self):
|
|
self.assertEqual(
|
|
ComplexNumber(0, 1) * ComplexNumber(0, 2), ComplexNumber(-2, 0)
|
|
)
|
|
|
|
def test_multiply_numbers_with_real_and_imaginary_part(self):
|
|
self.assertEqual(
|
|
ComplexNumber(1, 2) * ComplexNumber(3, 4), ComplexNumber(-5, 10)
|
|
)
|
|
|
|
# Division
|
|
|
|
def test_divide_purely_real_numbers(self):
|
|
self.assertAlmostEqual(
|
|
ComplexNumber(1, 0) / ComplexNumber(2, 0), ComplexNumber(0.5, 0)
|
|
)
|
|
|
|
def test_divide_purely_imaginary_numbers(self):
|
|
self.assertAlmostEqual(
|
|
ComplexNumber(0, 1) / ComplexNumber(0, 2), ComplexNumber(0.5, 0)
|
|
)
|
|
|
|
def test_divide_numbers_with_real_and_imaginary_part(self):
|
|
self.assertAlmostEqual(
|
|
ComplexNumber(1, 2) / ComplexNumber(3, 4), ComplexNumber(0.44, 0.08)
|
|
)
|
|
|
|
# Absolute value
|
|
|
|
def test_absolute_value_of_a_positive_purely_real_number(self):
|
|
self.assertEqual(abs(ComplexNumber(5, 0)), 5)
|
|
|
|
def test_absolute_value_of_a_negative_purely_real_number(self):
|
|
self.assertEqual(abs(ComplexNumber(-5, 0)), 5)
|
|
|
|
def test_absolute_value_of_a_purely_imaginary_number_with_positive_imaginary_part(
|
|
self,
|
|
):
|
|
self.assertEqual(abs(ComplexNumber(0, 5)), 5)
|
|
|
|
def test_absolute_value_of_a_purely_imaginary_number_with_negative_imaginary_part(
|
|
self,
|
|
):
|
|
self.assertEqual(abs(ComplexNumber(0, -5)), 5)
|
|
|
|
def test_absolute_value_of_a_number_with_real_and_imaginary_part(self):
|
|
self.assertEqual(abs(ComplexNumber(3, 4)), 5)
|
|
|
|
# Complex conjugate
|
|
|
|
def test_conjugate_a_purely_real_number(self):
|
|
self.assertEqual(ComplexNumber(5, 0).conjugate(), ComplexNumber(5, 0))
|
|
|
|
def test_conjugate_a_purely_imaginary_number(self):
|
|
self.assertEqual(ComplexNumber(0, 5).conjugate(), ComplexNumber(0, -5))
|
|
|
|
def test_conjugate_a_number_with_real_and_imaginary_part(self):
|
|
self.assertEqual(ComplexNumber(1, 1).conjugate(), ComplexNumber(1, -1))
|
|
|
|
# Complex exponential function
|
|
|
|
def test_euler_s_identity_formula(self):
|
|
self.assertAlmostEqual(ComplexNumber(0, math.pi).exp(), ComplexNumber(-1, 0))
|
|
|
|
def test_exponential_of_0(self):
|
|
self.assertAlmostEqual(ComplexNumber(0, 0).exp(), ComplexNumber(1, 0))
|
|
|
|
def test_exponential_of_a_purely_real_number(self):
|
|
self.assertAlmostEqual(ComplexNumber(1, 0).exp(), ComplexNumber(math.e, 0))
|
|
|
|
def test_exponential_of_a_number_with_real_and_imaginary_part(self):
|
|
self.assertAlmostEqual(
|
|
ComplexNumber(math.log(2), math.pi).exp(), ComplexNumber(-2, 0)
|
|
)
|
|
|
|
def test_exponential_resulting_in_a_number_with_real_and_imaginary_part(self):
|
|
self.assertAlmostEqual(
|
|
ComplexNumber(math.log(2) / 2, math.pi / 4).exp(), ComplexNumber(1, 1)
|
|
)
|
|
|
|
# Operations between real numbers and complex numbers
|
|
|
|
def test_add_real_number_to_complex_number(self):
|
|
self.assertEqual(ComplexNumber(1, 2) + 5, ComplexNumber(6, 2))
|
|
|
|
def test_add_complex_number_to_real_number(self):
|
|
self.assertEqual(5 + ComplexNumber(1, 2), ComplexNumber(6, 2))
|
|
|
|
def test_subtract_real_number_from_complex_number(self):
|
|
self.assertEqual(ComplexNumber(5, 7) - 4, ComplexNumber(1, 7))
|
|
|
|
def test_subtract_complex_number_from_real_number(self):
|
|
self.assertEqual(4 - ComplexNumber(5, 7), ComplexNumber(-1, -7))
|
|
|
|
def test_multiply_complex_number_by_real_number(self):
|
|
self.assertEqual(ComplexNumber(2, 5) * 5, ComplexNumber(10, 25))
|
|
|
|
def test_multiply_real_number_by_complex_number(self):
|
|
self.assertEqual(5 * ComplexNumber(2, 5), ComplexNumber(10, 25))
|
|
|
|
def test_divide_complex_number_by_real_number(self):
|
|
self.assertAlmostEqual(ComplexNumber(10, 100) / 10, ComplexNumber(1, 10))
|
|
|
|
def test_divide_real_number_by_complex_number(self):
|
|
self.assertAlmostEqual(5 / ComplexNumber(1, 1), ComplexNumber(2.5, -2.5))
|
|
|
|
# Additional tests for this track
|
|
|
|
def test_equality_of_complex_numbers(self):
|
|
self.assertEqual(ComplexNumber(1, 2), ComplexNumber(1, 2))
|
|
|
|
def test_inequality_of_real_part(self):
|
|
self.assertNotEqual(ComplexNumber(1, 2), ComplexNumber(2, 2))
|
|
|
|
def test_inequality_of_imaginary_part(self):
|
|
self.assertNotEqual(ComplexNumber(1, 2), ComplexNumber(1, 1))
|