"""Tests for distutils.command.clean.""" import os from distutils.command.clean import clean from distutils.tests import support class TestClean(support.TempdirManager): def test_simple_run(self): pkg_dir, dist = self.create_dist() cmd = clean(dist) # let's add some elements clean should remove dirs = [ (d, os.path.join(pkg_dir, d)) for d in ( 'build_temp', 'build_lib', 'bdist_base', 'build_scripts', 'build_base', ) ] for name, path in dirs: os.mkdir(path) setattr(cmd, name, path) if name == 'build_base': continue for f in ('one', 'two', 'three'): self.write_file(os.path.join(path, f)) # let's run the command cmd.all = 1 cmd.ensure_finalized() cmd.run() # make sure the files where removed for _name, path in dirs: assert not os.path.exists(path), f'{path} was not removed' # let's run the command again (should spit warnings but succeed) cmd.all = 1 cmd.ensure_finalized() cmd.run()