"""Tests for distutils.command.install_data.""" | |
import os | |
import sys | |
import unittest | |
from distutils.command.install_lib import install_lib | |
from distutils.extension import Extension | |
from distutils.tests import support | |
from distutils.errors import DistutilsOptionError | |
from test.test_support import run_unittest | |
class InstallLibTestCase(support.TempdirManager, | |
support.LoggingSilencer, | |
support.EnvironGuard, | |
unittest.TestCase): | |
def test_finalize_options(self): | |
pkg_dir, dist = self.create_dist() | |
cmd = install_lib(dist) | |
cmd.finalize_options() | |
self.assertEqual(cmd.compile, 1) | |
self.assertEqual(cmd.optimize, 0) | |
# optimize must be 0, 1, or 2 | |
cmd.optimize = 'foo' | |
self.assertRaises(DistutilsOptionError, cmd.finalize_options) | |
cmd.optimize = '4' | |
self.assertRaises(DistutilsOptionError, cmd.finalize_options) | |
cmd.optimize = '2' | |
cmd.finalize_options() | |
self.assertEqual(cmd.optimize, 2) | |
def _setup_byte_compile(self): | |
pkg_dir, dist = self.create_dist() | |
cmd = install_lib(dist) | |
cmd.compile = cmd.optimize = 1 | |
f = os.path.join(pkg_dir, 'foo.py') | |
self.write_file(f, '# python file') | |
cmd.byte_compile([f]) | |
return pkg_dir | |
@unittest.skipIf(sys.dont_write_bytecode, 'byte-compile not enabled') | |
def test_byte_compile(self): | |
pkg_dir = self._setup_byte_compile() | |
if sys.flags.optimize < 1: | |
self.assertTrue(os.path.exists(os.path.join(pkg_dir, 'foo.pyc'))) | |
else: | |
self.assertTrue(os.path.exists(os.path.join(pkg_dir, 'foo.pyo'))) | |
def test_get_outputs(self): | |
pkg_dir, dist = self.create_dist() | |
cmd = install_lib(dist) | |
# setting up a dist environment | |
cmd.compile = cmd.optimize = 1 | |
cmd.install_dir = pkg_dir | |
f = os.path.join(pkg_dir, 'foo.py') | |
self.write_file(f, '# python file') | |
cmd.distribution.py_modules = [pkg_dir] | |
cmd.distribution.ext_modules = [Extension('foo', ['xxx'])] | |
cmd.distribution.packages = [pkg_dir] | |
cmd.distribution.script_name = 'setup.py' | |
# get_output should return 4 elements | |
self.assertTrue(len(cmd.get_outputs()) >= 2) | |
def test_get_inputs(self): | |
pkg_dir, dist = self.create_dist() | |
cmd = install_lib(dist) | |
# setting up a dist environment | |
cmd.compile = cmd.optimize = 1 | |
cmd.install_dir = pkg_dir | |
f = os.path.join(pkg_dir, 'foo.py') | |
self.write_file(f, '# python file') | |
cmd.distribution.py_modules = [pkg_dir] | |
cmd.distribution.ext_modules = [Extension('foo', ['xxx'])] | |
cmd.distribution.packages = [pkg_dir] | |
cmd.distribution.script_name = 'setup.py' | |
# get_input should return 2 elements | |
self.assertEqual(len(cmd.get_inputs()), 2) | |
def test_dont_write_bytecode(self): | |
# makes sure byte_compile is not used | |
pkg_dir, dist = self.create_dist() | |
cmd = install_lib(dist) | |
cmd.compile = 1 | |
cmd.optimize = 1 | |
old_dont_write_bytecode = sys.dont_write_bytecode | |
sys.dont_write_bytecode = True | |
try: | |
cmd.byte_compile([]) | |
finally: | |
sys.dont_write_bytecode = old_dont_write_bytecode | |
self.assertTrue('byte-compiling is disabled' in self.logs[0][1]) | |
def test_suite(): | |
return unittest.makeSuite(InstallLibTestCase) | |
if __name__ == "__main__": | |
run_unittest(test_suite()) |