refined usage and options for "py.cleanup":

py.cleanup     # remove "*.pyc" and "*$py.class" (jython) files
    py.cleanup -e .swp -e .cache # also remove files with these extensions
    py.cleanup -s  # remove "build" and "dist" directory next to setup.py files
    py.cleanup -d  # also remove empty directories
    py.cleanup -a  # synonym for "-s -d -e 'pip-log.txt'"
    py.cleanup -n  # dry run, only show what would be removed

--HG--
branch : trunk

testing/cmdline/test_cmdline.py

@@ -45,8 +45,10 @@ class TestPyCleanup:
         assert p.check()
         pyc = p.new(ext='pyc')
         pyc.ensure()
+        pyclass = p.new(basename=p.basename + '$py.class')
         result = testdir.runpybin("py.cleanup", tmpdir)
         assert not pyc.check()
+        assert not pyclass.check()
 
     def test_dir_remove_simple(self, testdir, tmpdir):
         subdir = tmpdir.mkdir("subdir")
@@ -59,3 +61,43 @@ class TestPyCleanup:
         result = testdir.runpybin("py.cleanup", tmpdir, '-d')
         assert result.ret == 0
         assert not subdir.check()
+
+    @py.test.mark.multi(opt=["-s"])
+    def test_remove_setup_simple(self, testdir, tmpdir, opt):
+        subdir = tmpdir.mkdir("subdir")
+        p = subdir.ensure("setup.py")
+        subdir.mkdir("build").ensure("hello", "world.py")
+        egg1 = subdir.mkdir("something.egg-info")
+        egg1.mkdir("whatever")
+        okbuild = subdir.mkdir("preserved1").mkdir("build")
+        egg2 = subdir.mkdir("preserved2").mkdir("other.egg-info")
+        subdir.mkdir("dist")
+        result = testdir.runpybin("py.cleanup", opt, subdir)
+        assert result.ret == 0
+        assert okbuild.check()      
+        assert egg1.check()
+        assert egg2.check()
+        assert subdir.join("preserved1").check()
+        assert subdir.join("preserved2").check()
+        assert not subdir.join("build").check()
+        assert not subdir.join("dist").check()
+
+    def test_remove_all(self, testdir, tmpdir):
+        tmpdir.ensure("setup.py")
+        tmpdir.ensure("build", "xyz.py")
+        tmpdir.ensure("dist", "abc.py")
+        piplog = tmpdir.ensure("preserved2", "pip-log.txt")
+        tmpdir.ensure("hello.egg-info")
+        setup = tmpdir.ensure("setup.py")
+        tmpdir.ensure("src/a/b")
+        x = tmpdir.ensure("src/x.py")
+        x2 = tmpdir.ensure("src/x.pyc")
+        x3 = tmpdir.ensure("src/x$py.class")
+        result = testdir.runpybin("py.cleanup", "-a", tmpdir)
+        assert result.ret == 0
+        assert len(tmpdir.listdir()) == 3
+        assert setup.check()
+        assert x.check()
+        assert not x2.check()
+        assert not x3.check()
+        assert not piplog.check()