Rakefile in simple_profiler-0.1.0 vs Rakefile in simple_profiler-0.1.1

@@ -1,19 +1,25 @@
-require "bundler/gem_tasks"
-require "rake/testtask"
+require 'bundler/gem_tasks'
+require 'rake/testtask'
 
 Rake::TestTask.new(:spec) do |t|
   t.libs << 'spec'
-  t.pattern = 'spec/**/*_spec.rb'
+  t.libs << 'lib'
+  t.pattern = ENV['DIR'] ? File.join(ENV['DIR'], '**', '*_spec.rb') : 'spec/**/*_spec.rb'
   t.verbose = false
   t.warning = false
+  t.loader = nil if ENV['TEST']
+  ENV['TEST'], ENV['LINE'] = ENV['TEST'].split(':') if ENV['TEST'] && !ENV['LINE']
+  t.options = ''
+  t.options << "--name=/#{ENV['NAME']}/ " if ENV['NAME']
+  t.options << "-l #{ENV['LINE']} " if ENV['LINE'] && ENV['TEST']
 end
 
+task default: :spec
+
 desc 'Console'
 task :console do
   require 'pry'
   require 'simple_profiler'
   ARGV.clear
   Pry.start
-end
-
-task default: :spec
+end
\ No newline at end of file