A&A for dataset and lazar

1 files changed, 9 insertions, 4 deletions

diff --git a/Rakefile b/Rakefile
index d8026a6..eb2d231 100644
--- a/Rakefile
+++ b/Rakefile
@@ -1,19 +1,24 @@
 require 'rubygems'
 require 'opentox-ruby'
 
+TEST_USER = "guest"
+TEST_PW   = "guest"
+
 task ARGV[0] do
   puts ARGV[0]+".rb"
   require "./"+ARGV[0]+".rb"
 end
 
 task :setup do
-  @@classification_training_dataset = OpenTox::Dataset.create_from_csv_file("data/hamster_carcinogenicity.csv")
-  @@regression_training_dataset = OpenTox::Dataset.create_from_csv_file("data/EPAFHM.csv")
+  @@subjectid = OpenTox::Authorization.authenticate(TEST_USER,TEST_PW) 
+  @@classification_training_dataset = OpenTox::Dataset.create_from_csv_file("data/hamster_carcinogenicity.csv", @@subjectid)
+  @@regression_training_dataset = OpenTox::Dataset.create_from_csv_file("data/EPAFHM.csv", @@subjectid)
 end
 
 task :teardown do
-  @@classification_training_dataset.delete
-  @@regression_training_dataset.delete
+  @@classification_training_dataset.delete(@@subjectid)
+  @@regression_training_dataset.delete(@@subjectid)
+  OpenTox::Authorization.logout(@@subjectid)
 end
 
 [:all, :feature, :dataset, :fminer, :lazar].each do |t|