fixed automated test to match new tweaks of pymas

Cloud User · Cloud User · commit 7f5312679ef6 · 2019-10-17T17:31:25.000Z
diff --git a/tests/integration/test_pymas.py b/tests/integration/test_pymas.py
@@ -184,7 +184,12 @@ def test_from_pickle(train_data, pickle_file):
                 rc = py.appendSrcLine('            raise _compile_error');
                 rc = py.appendSrcLine('        msg = ""');
                 rc = py.appendSrcLine('        import numpy as np');
-                rc = py.appendSrcLine('        result = obj.predict(np.array([SepalLength,SepalWidth,PetalLength,PetalWidth]).reshape((1, -1)))');
+                rc = py.appendSrcLine('');
+                rc = py.appendSrcLine('        inputarray = np.array([SepalLength,SepalWidth,PetalLength,PetalWidth]).reshape((1, -1))');
+                rc = py.appendSrcLine('        column=["CRIM","ZN","INDUS","CHAS","NOX","RM","AGE","DIS","RAD","TAX","PTRATIO","B","LSTAT"]');
+                rc = py.appendSrcLine('        import pandas as pd');
+                rc = py.appendSrcLine('        inputrun=pd.DataFrame(data=inputarray, columns=column)');
+                rc = py.appendSrcLine('        result = obj.predict(inputrun)');
                 rc = py.appendSrcLine('        if result.size == 1:');
                 rc = py.appendSrcLine('            result = np.asscalar(result)');
                 rc = py.appendSrcLine('    except Exception as e:');
