From fd42578dce78f4c5a530177426e6a2ea11fc95ed Mon Sep 17 00:00:00 2001 From: Ryan Snyder Date: Fri, 22 Jun 2018 23:50:21 -0400 Subject: tests: update for lifter, generator bugfixes --- suite/unit.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'suite/unit.py') 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(): -- cgit v1.3.1