def tearDown(self): pt.set_parallel_schedule(self.original_schedule[0], self.original_schedule[1]) pt.set_num_threads(self.original_threads)
def test_parallel_sched(self): pt.set_parallel_schedule("dynamic", 4) self.assertSequenceEqual(pt.get_parallel_schedule(), ("dynamic", 4)) pt.set_parallel_schedule("static", 1) self.assertSequenceEqual(pt.get_parallel_schedule(), ("static", 1))