MLR tests

1 files changed, 10 insertions, 0 deletions

diff --git a/lazar.rb b/lazar.rb
index b5f01e6..62a512b 100644
--- a/lazar.rb
+++ b/lazar.rb
@@ -80,6 +80,16 @@ class LazarTest < Test::Unit::TestCase
     cleanup
   end
 
+  def test_create_regression_prop_mlr_model
+    create_model :dataset_uri => @@regression_training_dataset.uri, :prediction_algorithm => "local_mlr_prop"
+    predict_compound  OpenTox::Compound.from_smiles("c1ccccc1NN")
+    assert_equal 0.262.round_to(3), @predictions.first.confidence(@compounds.first).round_to(3)
+    assert_equal 123, @predictions.first.neighbors(@compounds.first).size
+    assert_equal 131, @model.features.size
+    assert_equal 0.374.round_to(3), @predictions.first.value(@compounds.first).round_to(3)
+    cleanup
+  end
+
   def test_classification_model
     create_model :dataset_uri => @@classification_training_dataset.uri
     puts @model.uri