prime_factors_test.py

# These tests are auto-generated with test data from:
# https://github.com/exercism/problem-specifications/tree/main/exercises/prime-factors/canonical-data.json
# File last updated on 2023-07-19

import unittest

from prime_factors import (
    factors,
)


class PrimeFactorsTest(unittest.TestCase):
    def test_no_factors(self):
        self.assertEqual(factors(1), [])

    def test_prime_number(self):
        self.assertEqual(factors(2), [2])

    def test_another_prime_number(self):
        self.assertEqual(factors(3), [3])

    def test_square_of_a_prime(self):
        self.assertEqual(factors(9), [3, 3])

    def test_product_of_first_prime(self):
        self.assertEqual(factors(4), [2, 2])

    def test_cube_of_a_prime(self):
        self.assertEqual(factors(8), [2, 2, 2])

    def test_product_of_second_prime(self):
        self.assertEqual(factors(27), [3, 3, 3])

    def test_product_of_third_prime(self):
        self.assertEqual(factors(625), [5, 5, 5, 5])

    def test_product_of_first_and_second_prime(self):
        self.assertEqual(factors(6), [2, 3])

    def test_product_of_primes_and_non_primes(self):
        self.assertEqual(factors(12), [2, 2, 3])

    def test_product_of_primes(self):
        self.assertEqual(factors(901255), [5, 17, 23, 461])

    def test_factors_include_a_large_prime(self):
        self.assertEqual(factors(93819012551), [11, 9539, 894119])