#!/usr/bin/env python3 import os import sys try: import pefile except ImportError: if 'CI' in os.environ: raise # Skip the test if not on CI sys.exit(77) executable = sys.argv[1] expected = int(sys.argv[2]) actual = pefile.PE(executable).dump_dict()['OPTIONAL_HEADER']['Subsystem']['Value'] print('subsystem expected: %d, actual: %d' % (expected, actual)) sys.exit(0 if (expected == actual) else 1)