increased coverage

wesleybeckner · Nov 2, 2018 · 6f0be26 · 6f0be26
1 parent 6d24b49
commit 6f0be26
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/salty/tests/test_data_manipulation.py b/salty/tests/test_data_manipulation.py
@@ -25,7 +25,7 @@ def test_2_devmodel_to_array(self):
         return X_train, Y_train, X_test, Y_test
 
     def test_3_merge_duplicates(self):
-        data = salty.merge_duplicates(self.devmodel)
+        data = salty.merge_duplicates(self.devmodel, keep_descriptors=True)
         return data
 
     def test_4_assign_category(self):

diff --git a/salty/tests/test_iupac_smiles.py b/salty/tests/test_iupac_smiles.py
@@ -18,8 +18,13 @@ def test_1_check_data(self):
             df = salty.load_data(self.data_files[i])
             self.check_data(df)
 
+    def test_2_check_wrong_ion(selfs):
+        ion = 'stupid_nonsense_string'
+        salty.check_name(ion)
+
     def test_benchmark(self):
         salty.Benchmark.run(self.test_1_check_data)
+        salty.Benchmark.run(self.test_2_check_wrong_ion)
 
     def check_data(self, df):
         startTime = datetime.datetime.now()