def test_writeHeader(self): with mock.patch('sys.stdout', new_callable=StringIO) as m_stdout: PullRequestLinuxDriverMerge.write_header() self.assertEqual( '''-------------------------------------------------------------------------------- - - Begin: PullRequestLinuxDriver-Merge.py - -------------------------------------------------------------------------------- ''', m_stdout.getvalue())
def test_writeHeader(self): print( "BEGIN: Test PullRequestLinuxDriverMerge.py :: `write_header()`:") #m_stdout = StringIO() #with contextlib.redirect_stdout(m_stdout): with mock.patch('sys.stdout', new_callable=StringIO) as m_stdout: PRMerge.write_header() self.assertIn("Begin: PullRequestLinuxDriverMerge.py", m_stdout.getvalue()) print( "FINISH: Test PullRequestLinuxDriverMerge.py :: `write_header()`:") return