improve test coverage

Author: ea-rus

mindsdb_sdk/server.py

@@ -23,9 +23,6 @@ class Server(Project):
 
     """
 
-    def a(self):
-        ...
-
     def __init__(self, api):
         # server is also mindsdb project
         super().__init__(api, 'mindsdb')

tests/test_sdk.py

@@ -126,16 +126,16 @@ def check_model(self, model, database, mock_post):
             f'Finetune {model.project.name}.{model_name} FROM d1 (select a from t1)'
         )
 
-        model.retrain(query, options={ 'x': 2 })
+        model.retrain(query, options={'x': 2})
         check_sql_call(
             mock_post,
             f'RETRAIN {model.project.name}.{model_name} FROM {query.database} ({query.sql})  USING x=2'
         )
 
-        model.retrain('select a from t1', database='d1')
+        model.retrain('select a from t1', database='d1', engine='openai')
         check_sql_call(
             mock_post,
-            f'RETRAIN {model.project.name}.{model_name} FROM d1 (select a from t1)'
+            f'RETRAIN {model.project.name}.{model_name} FROM d1 (select a from t1) USING engine=\'openai\''
         )
 
         # describe
@@ -183,6 +183,7 @@ class Test(BaseFlow):
     def test_flow(self, mock_post, mock_put):
         # check local
         server = mindsdb_sdk.connect()
+        str(server)
 
         assert server.api.url == 'http://127.0.0.1:47334'
 
@@ -606,6 +607,7 @@ def test_flow(self, mock_post, mock_put):
         self.check_project(project, database)
 
         project = con.projects.create('proj1')
+        str(project)
         check_sql_call(
             mock_post, 'CREATE DATABASE proj1 WITH ENGINE = "mindsdb", PARAMETERS = {}')
         self.check_project(project, database)
@@ -762,6 +764,7 @@ def check_project_models(self, project, database, mock_post):
 
         models = project.models.list()
         model = models[0]  # Model object
+        str(model)
 
         assert model.name == 'm1'
         assert model.get_status() == 'complete'
@@ -791,7 +794,7 @@ def check_project_models(self, project, database, mock_post):
                 'window': 10,
                 'horizon': 2
             },
-            module = 'LightGBM',  # has to be in options
+            module='LightGBM',  # has to be in options
         )
         check_sql_call(
             mock_post,
@@ -911,6 +914,11 @@ def check_database(self, database, mock_post):
         table2 = database.tables.create('t2', query)
         check_sql_call(mock_post, f'create table {database.name}.t2 (select * from tbl1)')
 
+        # create with replace
+        database.tables.create('t2', query, replace=True)
+        check_sql_call(mock_post, f'create or replace table {database.name}.t2 (select * from tbl1)')
+
+
         assert table2.name == 't2'
         self.check_table(table2)
 
@@ -969,6 +977,7 @@ def check_project_jobs(self, project, mock_post):
         assert job.name == 'job1'
         assert job.query_str == 'select 1'
 
+        dir(project.jobs)
         job = project.jobs.job1
         str(job)
         assert job.name == 'job1'
@@ -1001,6 +1010,17 @@ def check_project_jobs(self, project, mock_post):
             call_stack_num=-2
         )
 
+        project.jobs.create(
+            name='job2',
+            query_str='retrain m1'
+        )
+
+        check_sql_call(
+            mock_post,
+            f"CREATE JOB job2 (retrain m1)",
+            call_stack_num=-2
+        )
+
         project.jobs.drop('job2')
 
         check_sql_call(