def testPrettyPrint(self):
""" Test that we can call the pretty print function. """
# Print to stdout.
original_sys_stdout = sys.stdout
try:
stream_1 = StringIO.StringIO()
sys.stdout = stream_1
# Print to stdout.
ref_str = "This is what we print"
prettyPrint(ref_str)
# Check.
if MPICommons.myRank() == 0:
ref_str = ref_str + "\n"
else:
ref_str = ""
self.assertEqual(stream_1.getvalue(), ref_str)
finally:
# Put the original stdout back.
sys.stdout = original_sys_stdout
# Print to another stream.
stream_2 = StringIO.StringIO()
ref_str = "This is what we print next time."
prettyPrint(ref_str, output=stream_2)
# Check.
if MPICommons.myRank() == 0:
ref_str = ref_str + "\n"
else:
ref_str = ""
self.assertEqual(stream_2.getvalue(), ref_str)
def testPrintHeader(self):
""" Test the print header function. """
# Print to stdout.
original_sys_stdout = sys.stdout
try:
stream_1 = StringIO.StringIO()
sys.stdout = stream_1
# Print to stdout.
printHeader()
# Check.
if MPICommons.myRank() == 0:
ref_str = """# -----------------------------------------------------------------------------
# KMCLib version 1.1.01
# Distributed under the GPLv3 license
# Copyright (C) 2012-2015 Mikael Leetmaa
# Developed by Mikael Leetmaa <*****@*****.**>
#
# This program is distributed in the hope that it will be useful
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# LICENSE and README files, and the source code, for details.
#
# You should have received a copy of the GNU General Public License version 3
# (GPLv3) along with this program. If not, see <http://www.gnu.org/licenses/>.
# -----------------------------------------------------------------------------
"""
else:
ref_str = ""
# Check.
self.assertEqual(ref_str, stream_1.getvalue())
finally:
# Put the original stdout back.
sys.stdout = original_sys_stdout
# Print to another stream.
stream_2 = StringIO.StringIO()
prettyPrint(ref_str, output=stream_2)
# Check.
self.assertTrue(ref_str in stream_2.getvalue())
