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
