diff options
Diffstat (limited to 'setuptools/_distutils/tests/test_cmd.py')
| -rw-r--r-- | setuptools/_distutils/tests/test_cmd.py | 42 |
1 files changed, 22 insertions, 20 deletions
diff --git a/setuptools/_distutils/tests/test_cmd.py b/setuptools/_distutils/tests/test_cmd.py index 6a771a11..f5104e1d 100644 --- a/setuptools/_distutils/tests/test_cmd.py +++ b/setuptools/_distutils/tests/test_cmd.py @@ -7,6 +7,7 @@ from distutils.cmd import Command from distutils.dist import Distribution from distutils.errors import DistutilsOptionError from distutils import debug +import pytest class MyCmd(Command): @@ -14,7 +15,7 @@ class MyCmd(Command): pass -class CommandTestCase(unittest.TestCase): +class TestCommand(unittest.TestCase): def setUp(self): dist = Distribution() self.cmd = MyCmd(dist) @@ -29,36 +30,34 @@ class CommandTestCase(unittest.TestCase): cmd.ensure_string_list('yes_string_list') cmd.ensure_string_list('yes_string_list2') - self.assertRaises( - DistutilsOptionError, cmd.ensure_string_list, 'not_string_list' - ) + with pytest.raises(DistutilsOptionError): + cmd.ensure_string_list('not_string_list') - self.assertRaises( - DistutilsOptionError, cmd.ensure_string_list, 'not_string_list2' - ) + with pytest.raises(DistutilsOptionError): + cmd.ensure_string_list('not_string_list2') cmd.option1 = 'ok,dok' cmd.ensure_string_list('option1') - self.assertEqual(cmd.option1, ['ok', 'dok']) + assert cmd.option1 == ['ok', 'dok'] cmd.option2 = ['xxx', 'www'] cmd.ensure_string_list('option2') cmd.option3 = ['ok', 2] - self.assertRaises(DistutilsOptionError, cmd.ensure_string_list, 'option3') + with pytest.raises(DistutilsOptionError): + cmd.ensure_string_list('option3') def test_make_file(self): cmd = self.cmd # making sure it raises when infiles is not a string or a list/tuple - self.assertRaises( - TypeError, cmd.make_file, infiles=1, outfile='', func='func', args=() - ) + with pytest.raises(TypeError): + cmd.make_file(infiles=1, outfile='', func='func', args=()) # making sure execute gets called properly def _execute(func, args, exec_msg, level): - self.assertEqual(exec_msg, 'generating out from in') + assert exec_msg == 'generating out from in' cmd.force = True cmd.execute = _execute @@ -79,7 +78,7 @@ class CommandTestCase(unittest.TestCase): cmd.dump_options() wanted = ["command options for 'MyCmd':", ' option1 = 1', ' option2 = 1'] - self.assertEqual(msgs, wanted) + assert msgs == wanted def test_ensure_string(self): cmd = self.cmd @@ -88,37 +87,40 @@ class CommandTestCase(unittest.TestCase): cmd.option2 = None cmd.ensure_string('option2', 'xxx') - self.assertTrue(hasattr(cmd, 'option2')) + assert hasattr(cmd, 'option2') cmd.option3 = 1 - self.assertRaises(DistutilsOptionError, cmd.ensure_string, 'option3') + with pytest.raises(DistutilsOptionError): + cmd.ensure_string('option3') def test_ensure_filename(self): cmd = self.cmd cmd.option1 = __file__ cmd.ensure_filename('option1') cmd.option2 = 'xxx' - self.assertRaises(DistutilsOptionError, cmd.ensure_filename, 'option2') + with pytest.raises(DistutilsOptionError): + cmd.ensure_filename('option2') def test_ensure_dirname(self): cmd = self.cmd cmd.option1 = os.path.dirname(__file__) or os.curdir cmd.ensure_dirname('option1') cmd.option2 = 'xxx' - self.assertRaises(DistutilsOptionError, cmd.ensure_dirname, 'option2') + with pytest.raises(DistutilsOptionError): + cmd.ensure_dirname('option2') def test_debug_print(self): cmd = self.cmd with captured_stdout() as stdout: cmd.debug_print('xxx') stdout.seek(0) - self.assertEqual(stdout.read(), '') + assert stdout.read() == '' debug.DEBUG = True try: with captured_stdout() as stdout: cmd.debug_print('xxx') stdout.seek(0) - self.assertEqual(stdout.read(), 'xxx\n') + assert stdout.read() == 'xxx\n' finally: debug.DEBUG = False |