improved enm import

1 files changed, 5 insertions, 1 deletions

diff --git a/test/nanoparticles.rb b/test/nanoparticles.rb
index 8a6836c..6f241ec 100644
--- a/test/nanoparticles.rb
+++ b/test/nanoparticles.rb
@@ -3,8 +3,12 @@ require_relative "setup.rb"
 class NanoparticleTest  < MiniTest::Test
 
   def test_import
-    Import::Enanomapper.import
+    dataset_ids = Import::Enanomapper.import
     assert_operator Nanoparticle.count , :>, 570, "Only #{Nanoparticle.count} nanoparticles imported"
+    assert_operator dataset_ids.size, :>, 8, "Only #{dataset_ids.size} bundles imported"
+    p dataset_ids.collect{|d| Dataset.find(d).name}
+    assert dataset_ids.collect{|d| Dataset.find(d).name}.include? ("NanoWiki")
+    assert dataset_ids.collect{|d| Dataset.find(d).name}.include? ("Protein Corona Fingerprinting Predicts the Cellular Interaction of Gold and Silver Nanoparticles")
   end
 
   def test_create_model