From d61f78093f4ddf03c27a2c8ae0bab9c1f10c80f5 Mon Sep 17 00:00:00 2001 From: "helma@in-silico.ch" Date: Tue, 30 Oct 2018 17:26:59 +0100 Subject: tests fixed --- test/regression-model.rb | 17 ++++++----------- 1 file changed, 6 insertions(+), 11 deletions(-) (limited to 'test/regression-model.rb') diff --git a/test/regression-model.rb b/test/regression-model.rb index 0104741..55c1c5b 100644 --- a/test/regression-model.rb +++ b/test/regression-model.rb @@ -173,17 +173,12 @@ class LazarRegressionTest < MiniTest::Test model = Model::Lazar.create training_dataset: training_dataset result = model.predict training_dataset assert_kind_of Dataset, result - puts result.to_csv - puts result.features - # TODO - # check prediction - # check prediction_interval - # check warnings/applicability domain - assert 3, result.features.size - assert 8, result.compounds.size - assert_equal ["true"], result.values(result.compounds.first, result.features[1]) - assert_equal [0.65], result.values(result.compounds.first, result.features[2]) - assert_equal [0], result.values(result.compounds.first, result.features[2]) # classification returns nil, check if + assert_equal 6, result.features.size + assert_equal 88, result.compounds.size + assert_equal [1.95], result.values(result.compounds.first, result.bioactivity_features[0]).collect{|v| v.round(2)} + assert_equal [1.37], result.values(result.compounds[6], result.bioactivity_features[0]).collect{|v| v.round(2)} + assert_equal [1.79], result.values(result.compounds[6], result.prediction_features[0]).collect{|v| v.round(2)} + assert_equal [1.84,1.73], result.values(result.compounds[7], result.bioactivity_features[0]).collect{|v| v.round(2)} end end -- cgit v1.2.3