def test_string_display(self):
capturedOutput = io.StringIO() # Create StringIO object
sys.stdout = capturedOutput # and redirect stdout.
CS.print_chain_acceptance_info(chain)
sys.stdout = sys.__stdout__ # Reset redirect.
self.assertTrue(isinstance(capturedOutput.getvalue(), str),
msg='Caputured string')
self.assertFalse('Results dictionary' in capturedOutput.getvalue(),
msg='Expect results dictionary not included')
def test_string_display_with_results(self):
capturedOutput = io.StringIO() # Create StringIO object
sys.stdout = capturedOutput # and redirect stdout.
CS.print_chain_acceptance_info(
chain, results=dict(
nsimu=5000,
iacce=np.array([200, 800])
))
sys.stdout = sys.__stdout__ # Reset redirect.
self.assertTrue(isinstance(capturedOutput.getvalue(), str),
msg='Caputured string')
self.assertTrue('Results dictionary' in capturedOutput.getvalue(),
msg='Expect results dictionary included')