def test_known_composites(self):
self.assertEqual(prob_3.prim_fact(4), [2, 2])
self.assertEqual(prob_3.prim_fact(6), [2, 3])
self.assertEqual(prob_3.prim_fact(9), [3, 3])
self.assertEqual(prob_3.prim_fact(10), [2, 5])
self.assertEqual(prob_3.prim_fact(12), [2, 2, 3])
self.assertEqual(prob_3.prim_fact(14), [2, 7])
self.assertEqual(prob_3.prim_fact(15), [3, 5])
def test_prime_number_input(self):
#behaves well when input is prime number
self.assertEqual(prob_3.prim_fact(2), [2])
self.assertEqual(prob_3.prim_fact(3), [3])
self.assertEqual(prob_3.prim_fact(5), [5])
self.assertEqual(prob_3.prim_fact(7), [7])
self.assertEqual(prob_3.prim_fact(11), [11])
self.assertEqual(prob_3.prim_fact(13), [13])
self.assertEqual(prob_3.prim_fact(17), [17])