test_strategy.py

#!/usr/bin/env python
"""
Tests for strategy.py
"""

import os
import subprocess
import sys

if sys.version_info < (2, 7):
    import unittest2 as unittest
else:
    import unittest


class StrategyTest(unittest.TestCase):

    def test_print_output(self):
        """
        Verifies the print output when strategy.py is executed.
        The expected_output is equivalent to the output on the command
        line when running 'python strategy.py'.
        """
        output = subprocess.check_output(["python", "strategy.py"])
        expected_output = os.linesep.join([
            'Strategy Example 0',
            'Strategy Example 1 from execute 1',
            'Strategy Example 2 from execute 2',
            ''
        ])
        # byte representation required due to EOF returned subprocess
        expected_output_as_bytes = expected_output.encode(encoding='UTF-8')
        self.assertEqual(output, expected_output_as_bytes)

if __name__ == "__main__":
    unittest.main()