Rakefile in ruby-dnn-0.14.3 vs Rakefile in ruby-dnn-0.15.0

@@ -3,11 +3,11 @@
 
 Rake::TestTask.new(:test) do |t|
   t.libs << "test"
   t.libs << "ext"
   t.libs << "lib"
-  t.test_files = FileList["test/*_test.rb"]
+  t.test_files = FileList["test/*_test.rb", "test/layers_test/*_test.rb"]
 end
 
 task :build_cifar_loader do
   sh "cd ext/cifar_loader; ruby extconf.rb; make"
 end
@@ -25,9 +25,11 @@
 end
 
 task :default => [:test, :build_cifar_loader, :build_rb_stb_image]
 
 task :doc do
-  src_list = Dir["lib/dnn/core/*.rb"]
+  src_list = Dir["lib/dnn.rb"]
+  src_list += Dir["lib/dnn/core/*.rb"]
+  src_list += Dir["lib/dnn/core/layers/*.rb"]
   src_list += Dir["lib/dnn/*.rb"]
   sh "yardoc #{src_list.join(' ')}"
 end