def testNoMultiproc(self): """Tests that NUMBER_OF_PROCESSORS is used.""" with ou.change_environ('NUMBER_OF_PROCESSORS', '13'): self.assertEqual(13, pu.cpu_count(object()))
def testWithMultiprocessing(self): """Tests that multiprocessing.cpu_count is used.""" with mock.patch('multiprocessing.cpu_count', lambda: 11): self.assertEqual(11, pu.cpu_count())