fixing tests

Author: ea-rus

mindsdb_sdk/models.py

@@ -341,6 +341,7 @@ def describe(self, type: str = None) -> Union[pd.DataFrame, Query]:
         identifier = self._get_identifier()
         if type is not None:
             identifier.parts.append(type)
+            identifier.is_quoted.append(False)
         ast_query = Describe(identifier)
 
         sql = ast_query.to_string()

tests/test_sdk.py

@@ -107,11 +107,11 @@ def check_model(self, model, database, mock_post):
         assert call_args[1]['json']['params'] == params
 
         # check prediction
-        assert (pred_df == pd.DataFrame(data_out)).all().bool()
+        assert pred_df.equals(pd.DataFrame(data_out))
 
         # predict using dict
-        pred_df = model.predict({ 'a': 1 })
-        assert (pred_df == pd.DataFrame(data_out)).all().bool()
+        pred_df = model.predict({'a': 1})
+        assert pred_df.equals(pd.DataFrame(data_out))
 
         # using  deferred query
         response_mock(mock_post, pd.DataFrame(data_out))  # will be used sql/query
@@ -121,15 +121,15 @@ def check_model(self, model, database, mock_post):
 
         check_sql_call(mock_post,
                        f'SELECT m.* FROM (SELECT * FROM {query.database} (select a from t1)) AS t JOIN {model.project.name}.{model_name} AS m USING x="1"')
-        assert (pred_df == pd.DataFrame(data_out)).all().bool()
+        assert pred_df.equals(pd.DataFrame(data_out))
 
         # using table
         table0 = database.tables.tbl0
         pred_df = model.predict(table0)
 
         check_sql_call(mock_post,
                        f'SELECT m.* FROM (SELECT * FROM {table0.db.name}.tbl0) AS t JOIN {model.project.name}.{model_name} AS m')
-        assert (pred_df == pd.DataFrame(data_out)).all().bool()
+        assert pred_df.equals(pd.DataFrame(data_out))
 
 
         # time series prediction
@@ -138,7 +138,7 @@ def check_model(self, model, database, mock_post):
 
         check_sql_call(mock_post,
                        f'SELECT m.* FROM (SELECT * FROM {query.database} (select * from t1 where type="house" and saledate>latest)) as t JOIN {model.project.name}.{model_name} AS m')
-        assert (pred_df == pd.DataFrame(data_out)).all().bool()
+        assert pred_df.equals(pd.DataFrame(data_out))
 
         # -----------  model managing  --------------
         response_mock(
@@ -524,7 +524,7 @@ def check_database(self, database, mock_post):
 
         check_sql_call(mock_post, sql)
 
-        assert (data == result).all().bool()
+        assert data.equals(result)
 
         # test tables
         response_mock(mock_post, pd.DataFrame([{'name': 't1'}]))
@@ -990,7 +990,7 @@ def check_database(self, database, mock_post):
 
         check_sql_call(mock_post, sql)
 
-        assert (data == result).all().bool()
+        assert data.equals(result)
 
         # test tables
         response_mock(mock_post, pd.DataFrame([{'name': 't1'}]))