diff --git a/booltest/booltest_main.py b/booltest/booltest_main.py index cb2e02b5..85b04f11 100644 --- a/booltest/booltest_main.py +++ b/booltest/booltest_main.py @@ -737,7 +737,7 @@ def load_input_objects(self): :return: """ for file in self.args.files: - io = common.FileInputObject(fname=file) + io = common.FileInputObject(fname=file, fmode='rb') io.check() self.input_objects.append(io) diff --git a/booltest/common.py b/booltest/common.py index 5805ca8c..397b15af 100644 --- a/booltest/common.py +++ b/booltest/common.py @@ -367,13 +367,21 @@ class FileInputObject(InputObject): File input object - reading from the file """ def __init__(self, fname, *args, **kwargs): + """ + File reading input object + :param fname: + :param fmode: + :param args: + :param kwargs: + """ super(FileInputObject, self).__init__(*args, **kwargs) self.fname = fname + self.fmode = kwargs.get('fmode', 'r') self.fh = None def __enter__(self): super(FileInputObject, self).__enter__() - self.fh = open(self.fname, 'r') + self.fh = open(self.fname, self.fmode) def __exit__(self, exc_type, exc_val, exc_tb): super(FileInputObject, self).__exit__(exc_type, exc_val, exc_tb)