def testUnknownRaises(self): """Tests that a SystemError will be raised if the cpu count is indeterminable.""" with ou.change_environ('NUMBER_OF_PROCESSORS', None): self.assertRaises(SystemError, pu.cpu_count, object())
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 testDeletesIfItIsNew(self): self.assertNotIn('blah', os.environ) with osutils.change_environ('blah', 'f'): self.assertEqual(os.environ['blah'], 'f') self.assertNotIn('blah', os.environ)
def testReentrancy(self): # Will lock if not reentrant with osutils.change_environ('blah', 'f'): with osutils.change_environ('spam', 'e'): pass
def _changeAndTest(self, doRaise=False): with osutils.change_environ('TESTVALUE', 'XYZ'): self.assertEqual(os.environ['TESTVALUE'], 'XYZ') if doRaise: raise SystemError