def test_caching_array(self): self.factorManager = FactorManager(cache=True) input_list = [10, 5, 2, 20] return_list = self.factorManager.return_factor(input_list) expected_return = {10: [5, 2], 5: [], 2: [], 20: [10, 5, 2]} self.assertEqual(expected_return, return_list) return_list = self.factorManager.return_factor(input_list) self.assertEqual(expected_return, return_list)
def setUp(self): self.factorManager = FactorManager()