From 84e73eb2bf4f014f2f59d30c6b81cb0749d2faf4 Mon Sep 17 00:00:00 2001
From: gebele <gebele@in-silico.ch>
Date: Thu, 22 Oct 2015 11:35:32 +0000
Subject: added files and test for create dataset from file (smiles/inchi)
 without features, used for batch prediction

---
 test/dataset.rb | 19 ++++++++++---------
 1 file changed, 10 insertions(+), 9 deletions(-)

(limited to 'test/dataset.rb')

diff --git a/test/dataset.rb b/test/dataset.rb
index 47a6c25..4f1e885 100644
--- a/test/dataset.rb
+++ b/test/dataset.rb
@@ -12,15 +12,16 @@ class DatasetTest < MiniTest::Test
     d1.delete
   end
 
-  #TODO
-  def test_create_without_features_csv
-    d = Dataset.from_csv_file File.join(DATA_DIR,"batch_prediction.csv")
-    assert_equal Dataset, d.class
-    refute_nil d.id
-    dataset = Dataset.find d.id
-    #p dataset.compounds
-    assert_equal 24, d.compounds.size.to_i
-    d.delete
+  def test_create_without_features_smiles_and_inchi
+    ["smiles", "inchi"].each do |type|
+      d = Dataset.from_csv_file File.join(DATA_DIR,"batch_prediction_#{type}_small.csv")
+      assert_equal Dataset, d.class
+      refute_nil d.id
+      dataset = Dataset.find d.id
+      #p dataset.compounds
+      assert_equal 3, d.compounds.size.to_i
+      d.delete
+    end
   end
 
   def test_create_empty
-- 
cgit v1.2.3