def test_results_threading(self): def test_func(text): time.sleep(2) return len(text) p_helper = ParaHelper(test_func) start = time.time() p_helper.add_task("first") p_helper.add_task("second") p_helper.add_task("third") p_helper.add_task("fourth") p_helper.add_task("fifth") p_helper.add_task("sixth") p_helper.add_task("seventh") p_helper.add_task("eight") results = p_helper.get_all_results() end = time.time() expected = [ ["first", 5], ["second", 6], ["third", 5], ["fourth", 6], ["fifth", 5], ["sixth", 5], ["seventh", 7], ["eight", 5], ] self.assertCountEqual(expected, results) self.assertTrue(end - start < 4)
def test_results_processing(self): a = TestClass() p_helper = ParaHelper(a.test_func) p_helper.add_task("first") p_helper.add_task("second") results = p_helper.get_all_results() self.assertCountEqual([["first", 5], ["second", 6]], results)