diff options
Diffstat (limited to 'suite/unit.py')
| -rw-r--r-- | suite/unit.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/suite/unit.py b/suite/unit.py index fe2c4b8c..9cbed511 100644 --- a/suite/unit.py +++ b/suite/unit.py @@ -71,7 +71,7 @@ class TestBinaryNinjaAPI(unittest.TestCase): # Python 2 does not have the encodings option self.oracle_test_data = pickle.load(open(os.path.join("suite", "oracle.pkl"), "rb"), encoding='charmap') except TypeError: - self.oracle_test_data = pickle.load(open(os.path.join("suite", "oracle.pkl"), "r")) + self.oracle_test_data = pickle.load(open(os.path.join("suite", "oracle.pkl"), "rb")) self.verifybuilder = testcommon.VerifyBuilder("suite/binaries/test_corpus") def run_binary_test(self, testfile): @@ -85,7 +85,7 @@ class TestBinaryNinjaAPI(unittest.TestCase): # Python 2 does not have the encodings option binary_oracle = pickle.load(open(testname + ".pkl", "rb"), encoding='charmap') except TypeError: - binary_oracle = pickle.load(open(testname + ".pkl", "r")) + binary_oracle = pickle.load(open(testname + ".pkl", "rb")) test_builder = testcommon.BinaryViewTestBuilder(testname, "suite/binaries/test_corpus") for method in test_builder.methods(): |