def tearDown(self): pt.set_parallel_schedule(self.original_schedule[0], self.original_schedule[1]) pt.set_num_threads(self.original_threads)
def test_get_and_set_threads(self): self.assertEqual(pt.get_num_threads(), 1) pt.set_num_threads(4) self.assertEqual(pt.get_num_threads(), 4)