summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--test/prediction_models.rb2
1 files changed, 2 insertions, 0 deletions
diff --git a/test/prediction_models.rb b/test/prediction_models.rb
index b4ad415..001ebcd 100644
--- a/test/prediction_models.rb
+++ b/test/prediction_models.rb
@@ -15,6 +15,8 @@ class PredictionModelTest < MiniTest::Test
[:endpoint,:species,:source].each do |p|
refute_empty pm[p]
end
+ assert pm.classification?
+ refute pm.regression?
assert pm.crossvalidation.accuracy > 0.8
prediction = pm.predict Compound.from_smiles("CCCC(NN)C")
assert_equal "true", prediction[:value]