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