Fixed error in test_create unit test

This commit is contained in:
Rob McMullen 2018-11-11 12:40:28 -08:00
parent 7ce7d0349a
commit 0abcf2d929
1 changed files with 2 additions and 2 deletions

View File

@ -41,9 +41,9 @@ class BaseCreateTest:
try:
_ = issubclass(errors.AtrError, expected)
with pytest.raises(errors.InvalidBinaryFile) as e:
file_data, filetype = image.create_executable_file_image(segments, run_addr)
file_data, filetype = image.create_executable_file_image(sample_file, segments, run_addr)
except TypeError:
file_data, filetype = image.create_executable_file_image(segments, run_addr)
file_data, filetype = image.create_executable_file_image(sample_file, segments, run_addr)
print(image)
print(file_data, filetype)
assert len(file_data) == expected