def test_dump_binary(self): stream = StringIO() BinaryLoader.dump_binary(self.program, stream) stream.seek(0) newbinary = load_binary_file('the same file dumped', stream) eq_(newbinary, self.program)
def run(args=None): parser = argparse.ArgumentParser(description='TTK-91 Assembler') parser.add_argument('inputfile', type=argparse.FileType('r')) parser.add_argument('outputfile', type=argparse.FileType('w'), nargs='?', default=sys.stdout) res = parser.parse_args(args) if res: binary = Assembler(res.inputfile.name, res.inputfile.read()).build_binary() if isinstance(binary, list): for err in binary: print err else: BinaryLoader.dump_binary(binary, res.outputfile)