class PrimeFactorsTest(unittest.TestCase): def setUp(self): self.PrimeFactors = PrimeFactors() def test_one(self): self.assertEqual( [], self.PrimeFactors.generate(1) ) def test_two(self): self.assertEqual( [2], self.PrimeFactors.generate(2) ) def test_three(self): self.assertEqual( [3], self.PrimeFactors.generate(3) ) def test_four(self): self.assertEqual( [2,2], self.PrimeFactors.generate(4) ) def test_six(self): self.assertEqual( [2,3], self.PrimeFactors.generate(6) ) def test_eight(self): self.assertEqual( [2,2,2], self.PrimeFactors.generate(8) ) def test_nine(self): self.assertEqual( [3,3], self.PrimeFactors.generate(9) )
def setUp(self): self.PrimeFactors = PrimeFactors()