FilesExpand file tree

 master

/

test_strategy.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

36 lines (30 loc) · 987 Bytes

 master

/

test_strategy.py

Copy path

Top

File metadata and controls

• Code
• Blame

36 lines (30 loc) · 987 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

#!/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()