test: update for changes to viya & pandas

Author: jlwalke2

tests/integration/test_model_repository.py

@@ -54,7 +54,9 @@ def test_copy_astore(self):
         """Copy the ASTORE to filesystem for MAS"""
         job = mr.copy_analytic_store(self.MODEL_NAME)
 
-        assert job.state == "pending"
+        # Later versions of Viya 4 seem to have dropped the "state" attribute.
+        if hasattr(job, "state"):
+            assert job.state == "pending"
 
     def test_model_publish(self, cache):
         """Publish the imported model to MAS"""

tests/integration/test_tasks.py

@@ -34,8 +34,7 @@ def sklearn_logistic_model():
     iris = pd.DataFrame(raw.data, columns=raw.feature_names)
     iris = iris.join(pd.DataFrame(raw.target))
     iris.columns = ["SepalLength", "SepalWidth", "PetalLength", "PetalWidth", "Species"]
-    iris["Species"] = iris["Species"].astype("category")
-    iris.Species.cat.categories = raw.target_names
+    iris["Species"] = iris["Species"].astype("category").cat.rename_categories(raw.target_names)
 
     with warnings.catch_warnings():
         warnings.simplefilter("ignore")
@@ -205,7 +204,7 @@ def test_register_model(self, sklearn_linear_model):
         assert "prediction" == model.function.lower()
         assert "Linear regression" == model.algorithm
         assert "Python" == model.trainCodeType
-        assert "ds2MultiType" == model.scoreCodeType
+        assert model.scoreCodeType in ("ds2MultiType", "python")
 
         assert len(model.inputVariables) == 13
         assert len(model.outputVariables) == 1

tests/integration/test_write_json_files.py

@@ -19,7 +19,8 @@ def _classification_model(data, target):
     from sklearn.model_selection import train_test_split
     from sklearn.tree import DecisionTreeClassifier
 
-    data = pd.get_dummies(data, drop_first=True).fillna(data.mean())
+    data = pd.get_dummies(data, drop_first=True)
+    data.fillna(data.mean(), inplace=True)
     x_train, x_test, y_train, y_test = train_test_split(
         data.drop(columns=target), data[target], test_size=0.3
     )