diff --git a/setup.py b/setup.py index 73ae4a6..7db697f 100644 --- a/setup.py +++ b/setup.py @@ -48,9 +48,12 @@ class CMakeBuild(build_ext): cmake_args += ['-DCMAKE_BUILD_TYPE=' + cfg] build_args += ['--', '-j2'] + env = os.environ.copy() + env['CXXFLAGS'] = '{} -DVERSION_INFO=\\"{}\\"'.format(env.get('CXXFLAGS', ''), + self.distribution.get_version()) if not os.path.exists(self.build_temp): os.makedirs(self.build_temp) - subprocess.check_call(['cmake', ext.sourcedir] + cmake_args, cwd=self.build_temp) + subprocess.check_call(['cmake', ext.sourcedir] + cmake_args, cwd=self.build_temp, env=env) subprocess.check_call(['cmake', '--build', '.'] + build_args, cwd=self.build_temp) setup( diff --git a/tests/test.py b/tests/test.py index 30997e4..cea38c4 100644 --- a/tests/test.py +++ b/tests/test.py @@ -1,5 +1,5 @@ import cmake_example as m -assert m.__version__ == "dev" +assert m.__version__ == '0.0.1' assert m.add(1, 2) == 3 assert m.subtract(1, 2) == -1