forked from Archive/PX4-Autopilot
51 lines
1.4 KiB
Python
51 lines
1.4 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
"""
|
||
|
|
This runs a command and compares output to a known file over
|
||
|
|
a given line range.
|
||
|
|
"""
|
||
|
|
from __future__ import print_function
|
||
|
|
import subprocess
|
||
|
|
import argparse
|
||
|
|
import os
|
||
|
|
|
||
|
|
|
||
|
|
#pylint: disable=invalid-name
|
||
|
|
parser = argparse.ArgumentParser(description='Process some integers.')
|
||
|
|
parser.add_argument('--command', required=True)
|
||
|
|
parser.add_argument('--stdin', required=True)
|
||
|
|
parser.add_argument('--stdout', required=True)
|
||
|
|
parser.add_argument('--check', required=True)
|
||
|
|
parser.add_argument('--start', default=0)
|
||
|
|
parser.add_argument('--stop', default=-1)
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
d = os.path.dirname(args.stdout)
|
||
|
|
if not os.path.exists(d):
|
||
|
|
os.makedirs(d)
|
||
|
|
|
||
|
|
with open(args.stdout, 'w') as outfile:
|
||
|
|
with open(args.stdin, 'r') as infile:
|
||
|
|
proc = subprocess.Popen(
|
||
|
|
args.command, stdout=outfile, stdin=infile)
|
||
|
|
proc.communicate()
|
||
|
|
|
||
|
|
i_start = int(args.start)
|
||
|
|
i_stop = int(args.stop)
|
||
|
|
|
||
|
|
with open(args.stdout, 'r') as outfile:
|
||
|
|
out_contents = file.readlines(outfile)
|
||
|
|
out_contents = "".join(out_contents[i_start:i_stop])
|
||
|
|
|
||
|
|
with open(args.check, 'r') as checkfile:
|
||
|
|
check_contents = file.readlines(checkfile)
|
||
|
|
check_contents = "".join(check_contents[i_start:i_stop])
|
||
|
|
|
||
|
|
if (out_contents != check_contents):
|
||
|
|
print("output:\n", out_contents)
|
||
|
|
print("check:\n", check_contents)
|
||
|
|
exit(1)
|
||
|
|
|
||
|
|
exit(0)
|
||
|
|
|
||
|
|
# vim: set et ft=python fenc= ff=unix sts=4 sw=4 ts=4 :
|