diff options
Diffstat (limited to 'test/lazar-fminer.rb')
-rw-r--r-- | test/lazar-fminer.rb | 23 |
1 files changed, 9 insertions, 14 deletions
diff --git a/test/lazar-fminer.rb b/test/lazar-fminer.rb index 11fef72..6c0ee41 100644 --- a/test/lazar-fminer.rb +++ b/test/lazar-fminer.rb @@ -3,18 +3,16 @@ require_relative "setup.rb" class LazarFminerTest < MiniTest::Test def test_lazar_fminer - training_dataset = OpenTox::Dataset.from_csv_file File.join(DATA_DIR,"hamster_carcinogenicity.csv") - feature_dataset = OpenTox::Algorithm::Fminer.bbrc(:dataset => training_dataset) - #p feature_dataset - model = OpenTox::Model::Lazar.create training_dataset, feature_dataset + training_dataset = Dataset.from_csv_file File.join(DATA_DIR,"hamster_carcinogenicity.csv") + feature_dataset = Algorithm::Fminer.bbrc(training_dataset) + model = Model::Lazar.create training_dataset, feature_dataset #feature_dataset = OpenTox::Dataset.find model.feature_dataset_id - p model assert_equal training_dataset.compounds.size, feature_dataset.compounds.size assert_equal 54, feature_dataset.features.size - #feature_dataset.data_entries.each do |e| - #assert_equal e.size, feature_dataset.features.size - #end - assert_equal '[#6&A]-[#6&A]-[#6&A]=[#6&A]', feature_dataset.features.first.smarts + feature_dataset.data_entries.each do |e| + assert_equal e.size, feature_dataset.features.size + end + assert_equal 'C-C-C=C', feature_dataset.features.first.smarts [ { :compound => OpenTox::Compound.from_inchi("InChI=1S/C6H6/c1-2-4-6-5-3-1/h1-6H"), @@ -34,14 +32,11 @@ class LazarFminerTest < MiniTest::Test end # make a dataset prediction - compound_dataset = OpenTox::Dataset.from_sdf File.join(DATA_DIR,"EPAFHM.mini.csv") - #assert_equal compound_dataset.uri.uri?, true + compound_dataset = OpenTox::Dataset.from_csv_file File.join(DATA_DIR,"EPAFHM.mini.csv") prediction = model.predict :dataset => compound_dataset assert_equal compound_dataset.compounds, prediction.compounds - #prediction = OpenTox::Dataset.new prediction_uri - #assert_equal prediction.uri.uri?, true # cleanup - [dataset,model,feature_dataset,compound_dataset].each{|o| o.delete} + [training_dataset,model,feature_dataset,compound_dataset].each{|o| o.delete} end end |