Skip to content

Commit 07bb548

Browse files
committed
ignore byte string for comparison
1 parent ab28c2a commit 07bb548

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

tests/integration/test_pymas.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -106,6 +106,7 @@ def test_from_inline():
106106

107107

108108
def test_from_pickle(train_data, pickle_file):
109+
import re
109110
from sasctl.utils.pymas import PyMAS, from_pickle
110111

111112
X, y = train_data
@@ -139,7 +140,7 @@ def test_from_pickle(train_data, pickle_file):
139140
if rc then do;
140141
rc = py.appendSrcLine('try:');
141142
rc = py.appendSrcLine(' import pickle, base64');
142-
rc = py.appendSrcLine(' bytes = b"gANjc2tsZWFybi5saW5lYXJfbW9kZWwubG9naXN0aWMKTG9naXN0aWNSZWdyZXNzaW9uCnEAKYFxAX1xAihYBwAAAHBlbmFsdHlxA1gCAAAAbDJxBFgEAAAAZHVhbHEFiVgDAAAAdG9scQZHPxo24uscQy1YAQAAAENxB0c/8AAAAAAAAFgNAAAAZml0X2ludGVyY2VwdHEIiFgRAAAAaW50ZXJjZXB0X3NjYWxpbmdxCUsBWAwAAABjbGFzc193ZWlnaHRxCk5YDAAAAHJhbmRvbV9zdGF0ZXELTlgGAAAAc29sdmVycQxYBQAAAGxiZmdzcQ1YCAAAAG1heF9pdGVycQ5LZFgLAAAAbXVsdGlfY2xhc3NxD1gLAAAAbXVsdGlub21pYWxxEFgHAAAAdmVyYm9zZXERSwBYCgAAAHdhcm1fc3RhcnRxEolYBgAAAG5fam9ic3ETSwFYCAAAAGNsYXNzZXNfcRRjbnVtcHkuY29yZS5tdWx0aWFycmF5Cl9yZWNvbnN0cnVjdApxFWNudW1weQpuZGFycmF5CnEWSwCFcRdDAWJxGIdxGVJxGihLAUsDhXEbY251bXB5CmR0eXBlCnEcWAIAAABPOHEdSwBLAYdxHlJxHyhLA1gBAAAAfHEgTk5OSv////9K/////0s/dHEhYoldcSIoWAYAAABzZXRvc2FxI1gKAAAAdmVyc2ljb2xvcnEkWAkAAAB2aXJnaW5pY2FxJWV0cSZiWAUAAABjb2VmX3EnaBVoFksAhXEoaBiHcSlScSooSwFLA0sEhnEraBxYAgAAAGY4cSxLAEsBh3EtUnEuKEsDWAEAAAA8cS9OTk5K/////0r/////SwB0cTBiiUNg7q0YrwEZ279NCwAVXMbuP7dX62b7JwTAIZOt8/Nf8b+SPbsd0xfhPx0uMT+6WdS/BvEkXs5Jyr/J0cJa2xHuvys0dzGSWry/RXRn9X6Z5L+wps1MmMwFQP59h9BwNABAcTF0cTJiWAoAAABpbnRlcmNlcHRfcTNoFWgWSwCFcTRoGIdxNVJxNihLAUsDhXE3aC6JQxjw3z6lNsMjQNZXvTklwQFAWTau838zKMBxOHRxOWJYBwAAAG5faXRlcl9xOmgVaBZLAIVxO2gYh3E8UnE9KEsBSwGFcT5oHFgCAAAAaTRxP0sASwGHcUBScUEoSwNoL05OTkr/////Sv////9LAHRxQmKJQwRfAAAAcUN0cURiWBAAAABfc2tsZWFybl92ZXJzaW9ucUVYBgAAADAuMTkuMXFGdWIu"');
143+
rc = py.appendSrcLine(' bytes = b"X"');
143144
rc = py.appendSrcLine(' obj = pickle.loads(base64.b64decode(bytes))');
144145
rc = py.appendSrcLine(' _compile_error = None');
145146
rc = py.appendSrcLine('except Exception as e:');
@@ -192,7 +193,9 @@ def test_from_pickle(train_data, pickle_file):
192193

193194
# Drop leading \n caused by multiline comment formatting
194195
result = p.score_code()
195-
assert target.lstrip('\n') == result
196+
197+
assert re.sub('bytes = b"[\w\d/\+]+"', 'bytes = b"X"', result) == \
198+
target.lstrip('\n')
196199

197200

198201
def test_from_pickle_stream(train_data, pickle_stream):

0 commit comments

Comments
 (0)