1import sys 2import os 3from distutils.core import setup, Extension 4 5USE_CPP = (os.environ.get('TEST_USE_CPP') == 'yes') 6 7# The C/C++ source is the same, but GCC recognizes the language by extension, 8# so use a symlink named 'test_py3c.cpp' for C++. 9# (there's also a gcc -x switch, but it needs to go before the filename; 10# I don't think setuptools allows that) 11if USE_CPP: 12 sources = ['test_py3c.cpp'] 13else: 14 sources = ['test_py3c.c'] 15 16extra_compile_args = [] 17extra_compile_args.extend(['-Werror', '-Wall']) 18if sys.version_info < (2, 7): 19 # XXX: The PyCapsule (2.6) shim tests aren't yet warning-free in GCC 20 extra_compile_args.extend(['-Wno-format']) 21 22test_py3c_module = Extension( 23 'test_py3c', 24 sources=sources, 25 include_dirs=['../include'], 26 extra_compile_args=extra_compile_args, 27) 28test_py3c_module.extra_compile_args=extra_compile_args 29 30setup_args = dict( 31 name='test_py3c', 32 version='0.0', 33 description = '', 34 ext_modules = [test_py3c_module] 35) 36 37if __name__ == '__main__': 38 setup(**setup_args) 39