Clean up tmp folder after running rake task

package_task.rb

require 'rake/packagetask'

task default: [:package]

Rake::PackageTask.new "package_name", :noversion do |package|
  package.need_zip = true
  package.package_files.include "images/*"
end

# Rake::PackageTask.new do |package|
#   package.name = "package_name"
#   package.version = :noversion
#   package.package_dir = "package_dir"
#   package.package_files.include "images/*"
#   package.need_zip = true
#   package.define
# end

Rakefile

require 'task/testtask'
require 'rake/clean'

task default: [:test]

Rake::TestTask.new do |task|
  task.pattern = "test/*_test.rb"
end

CLEAN.include 'tmp'

task(:test).enhance do
  task(:clean).invoke
end

Zakefile.rb

task default: [:test]

task :test do
  begin
    ruby "test/*_test.rb"
  ensure
    task(:clean).invoke
  end
end

task :clean do
  rm_r "tmp"
end