require "bundler/gem_tasks"
require "rake/testtask"

Rake::TestTask.new(:test) do |t|
  t.libs << "test"
  t.libs << "ext"
  t.libs << "lib"
  t.test_files = FileList["test/*_test.rb"]
end

task :build_dataset_loader do
  sh "cd ext/cifar10_loader; ruby extconf.rb; make"
end

task :build_image_io do
  sh "cd ext/rb_stb_image; ruby extconf.rb; make"
end

task :default => [:test, :build_dataset_loader, :build_image_io]

task :doc do
  src_list = Dir["lib/dnn/core/*.rb"]
  src_list += Dir["lib/dnn/*.rb"]
  sh "yardoc #{src_list.join(' ')}"
end