test_prime_factors.py

from unittest import TestCase

from prime_factors import generate_prime_factors


class TestGeneratePrimeFactors(TestCase):
    def test_one_should_have_empty_prime_factors_list(self):
        self.assertEqual(generate_prime_factors(1), [])

    def test_two_should_have_two_on_prime_factors_list(self):
        self.assertEqual(generate_prime_factors(2), [2])

    def test_three_should_have_three_on_prime_factors_list(self):
        self.assertEqual(generate_prime_factors(3), [3])

    def test_four_should_have_two_and_two_on_prime_factors_list(self):
        self.assertEqual(generate_prime_factors(4), [2, 2])

    def test_six_should_have_two_and_three_on_prime_factors_list(self):
        self.assertEqual(generate_prime_factors(6), [2, 3])

    def test_eight_should_have_two_and_two_and_two_on_prime_factors_list(self):
        self.assertEqual(generate_prime_factors(8), [2, 2, 2])

    def test_nine_should_have_three_and_three_on_prime_factors_list(self):
        self.assertEqual(generate_prime_factors(9), [3, 3])