tests

Author: pplonski

mljar/client/base.py

@@ -66,6 +66,6 @@ def _check_response_status(self, response):
             elif response.status_code == 400:
                 raise BadRequestException(response.content)
             elif response.status_code == 500:
-                raise MljarException('server error: ' +response.content)
+                raise MljarException('server error: ' +str(response.content))
             else:
                 response.raise_for_status()

mljar/client/project.py

@@ -1,6 +1,7 @@
 from .base import MljarHttpClient
 from ..model.project import Project
 from ..exceptions import NotFoundException, CreateProjectException
+from ..log import logger
 
 class ProjectClient(MljarHttpClient):
     '''
@@ -49,6 +50,7 @@ def delete_project(self, hid):
         '''
         Deletes project
         '''
+        logger.info('Remove project: %s' % hid)
         response = self.request("DELETE", '/'.join([self.url, hid]))
         return response.status_code == 204 or response.status_code == 200
 

tests/dataset_client_test.py

@@ -32,15 +32,15 @@ def tearDown(self):
 
 
     def test_get_datasests(self):
-        """
-        Get empty list of datasets in project.
-        """
+
+        #Get empty list of datasets in project.
+
         # get datasets
         datasets = DatasetClient(self.project.hid).get_datasets()
         self.assertEqual(datasets, [])
 
     def test_prepare_data(self):
-        """ Test _prepare_data method on numpy array data """
+        #Test _prepare_data method on numpy array data
         dc = DatasetClient(self.project.hid)
         samples = 100
         columns = 10
@@ -56,7 +56,7 @@ def test_prepare_data(self):
         self.assertTrue('attribute_10' in data.columns)
 
     def test_get_dataset_for_wrong_hid(self):
-        """ Get dataset for wrong hid should return None """
+        #Get dataset for wrong hid should return None
         dc = DatasetClient(self.project.hid)
         dataset = dc.get_dataset('some-wrong-hid')
         self.assertTrue(dataset is None)
@@ -82,7 +82,6 @@ def test_add_dataset_for_training(self):
         self.assertTrue('title' in str(my_dataset_2))
         self.assertTrue('file' in str(my_dataset_2))
 
-
     def test_add_dataset_for_prediction(self):
         # setup dataset client
         dc = DatasetClient(self.project.hid)
@@ -100,7 +99,6 @@ def test_add_dataset_for_prediction(self):
         self.assertEqual(my_dataset.hid, my_dataset_2.hid)
         self.assertEqual(my_dataset.title, my_dataset_2.title)
 
-
     def test_add_existing_dataset(self):
         # setup dataset client
         dc = DatasetClient(self.project.hid)
@@ -119,14 +117,12 @@ def test_add_existing_dataset(self):
         datasets = dc.get_datasets()
         self.assertEqual(len(datasets), init_datasets_cnt+1)
 
-
     def test_prepare_data_two_sources(self):
         dc = DatasetClient(self.project.hid)
         data_1, data_hash_1 = dc._prepare_data(self.X, self.y)
         data_2, data_hash_2 = dc._prepare_data(self.X, None)
         self.assertNotEqual(data_hash_1, data_hash_2)
 
-
     def test_prepare_data_two_sources_numpy(self):
         dc = DatasetClient(self.project.hid)
         data_1, data_hash_1 = dc._prepare_data(np.array(self.X), np.array(self.y))

tests/experiment_client_test.py

@@ -40,10 +40,10 @@ def setUp(self):
 
     def tearDown(self):
         # wait before clean, to have time to initialize models
-        #time.sleep(60)
+        time.sleep(60)
         # clean
-        #self.project_client.delete_project(self.project.hid)
-        pass
+        self.project_client.delete_project(self.project.hid)
+
 
     '''
     def test_create_with_kfold_cv(self):

tests/project_based_test.py

@@ -18,13 +18,13 @@ def clean_projects():
         project_client = ProjectClient()
         projects = project_client.get_projects()
         for proj in projects:
-            if proj.title.startswith('Test'):
+            if proj.title.startswith('Test') and proj.title.endswith(get_postfix()):
                 project_client.delete_project(proj.hid)
 
     @classmethod
     def setUpClass(cls):
         ProjectBasedTest.clean_projects()
 
-    #@classmethod
-    #def tearDownClass(cls):
-    #    ProjectBasedTest.clean_projects()
+    @classmethod
+    def tearDownClass(cls):
+        ProjectBasedTest.clean_projects()