more tests

pplonski · pplonski · commit 53afc33c5a0b · 2017-02-21T22:25:15.000+01:00
diff --git a/tests/mljar_test.py b/tests/mljar_test.py
@@ -9,6 +9,7 @@
 
 from mljar.client.project import ProjectClient
 from project_based_test import ProjectBasedTest
+from mljar.exceptions import BadValueException, IncorrectInputDataException
 from mljar import Mljar
 
 class MljarTest(ProjectBasedTest):
@@ -50,6 +51,29 @@ def test_basic_usage(self):
         score = self.mse(pred, self.y)
         self.assertTrue(score < 0.1)
 
+    def test_empty_project_title(self):
+        with self.assertRaises(BadValueException) as context:
+            model = Mljar(project = '', experiment = '')
+
+    def test_wrong_tuning_mode(self):
+        with self.assertRaises(BadValueException) as context:
+            model = Mljar(project = self.proj_title, experiment = self.expt_title,
+                            tuning_mode = 'Crazy')
+
+    def test_default_tuning_mode(self):
+        model = Mljar(project = self.proj_title, experiment = self.expt_title)
+        self.assertEqual(model.tuning_mode, 'Sport')
+
+    def test_wrong_input_dim(self):
+        with self.assertRaises(IncorrectInputDataException) as context:
+            model = Mljar(project = self.proj_title, experiment = self.expt_title)
+            samples = 100
+            columns = 10
+            X = np.random.rand(samples, columns)
+            y = np.random.choice([0,1], samples+1, replace = True)
+            model.fit(X, y)
+
+
     def test_non_wait_fit(self):
         '''
         Test the non wait fit.
