Update python test code to python 3

justinmeza · Oct 19, 2019 · 58ed46a · 58ed46a
1 parent 6762b72
commit 58ed46a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/cmake/AddLolTest.cmake b/cmake/AddLolTest.cmake
@@ -7,7 +7,7 @@ FUNCTION(ADD_LOL_TEST TEST_NAME)
     SET(ARG_LOLCODE ${CMAKE_CURRENT_SOURCE_DIR}/test.lol)
   ENDIF(NOT ARG_LOLCODE)
 
-  SET( TEST_COMMAND python ${CMAKE_SOURCE_DIR}/test/testDriver.py ${CMAKE_BINARY_DIR}/lci ${ARG_LOLCODE} )
+  SET( TEST_COMMAND python3 ${CMAKE_SOURCE_DIR}/test/testDriver.py ${CMAKE_BINARY_DIR}/lci ${ARG_LOLCODE} )
 
   IF(ARG_OUTPUT)
     LIST(APPEND TEST_COMMAND -o=${CMAKE_CURRENT_SOURCE_DIR}/${ARG_OUTPUT})

diff --git a/test/testDriver.py b/test/testDriver.py
@@ -9,8 +9,8 @@
 parser = argparse.ArgumentParser(description="Driver for lci tests")
 parser.add_argument('pathToLCI', help="The absolute path the the lci executable")
 parser.add_argument('lolcodeFile', help="The absolute path to the lolcode file to test")
-parser.add_argument('-o', '--outputFile', type=argparse.FileType('r'), default=None, help="The expected output")
-parser.add_argument('-i', '--inputFile', type=argparse.FileType('r'), default=None, help="File to be used as input")
+parser.add_argument('-o', '--outputFile', type=argparse.FileType('rb'), default=None, help="The expected output")
+parser.add_argument('-i', '--inputFile', type=argparse.FileType('rb'), default=None, help="File to be used as input")
 parser.add_argument('-e', '--expectError', action="store_true", help="Specify that an error should occur")
 parser.add_argument('-m', '--memCheck', action='store_true', help="Do a memory check")