254c728b144e998bd3408557fb27877b524dc906,mltsp/tests/test_flask_app.py,FlaskAppTestCase,test_prediction_proc,#FlaskAppTestCase#,1492

Before Change

                                        "TEMP_TEST01"}).run(conn)
            r.table("predictions").insert({"id": "TEMP_TEST01"}).run(conn)
            fa.prediction_proc(
                pjoin(cfg.UPLOAD_FOLDER, "TESTRUN_215153.dat"),
                "TEMP_TEST01", "TEMP_TEST01", "RandomForestClassifier",
                "TEMP_TEST01", "TEMP_TEST01",
                metadata_file=pjoin(cfg.UPLOAD_FOLDER,
                                    "TESTRUN_215153_metadata.dat"))

            entry = r.table("predictions").get("TEMP_TEST01").run(conn)
            pred_results_dict = entry
            classes = np.array([el[0] for el in
                                pred_results_dict["pred_results_dict"]["TESTRUN_215153"]],
                               dtype="U")
            assert all(c in ["class1", "class2", "class3"] for c in classes)

            assert all(key in pred_results_dict for key in \
                       ("ts_data_dict", "features_dict"))
            for fpath in [pjoin(cfg.UPLOAD_FOLDER, "TESTRUN_215153.dat"),
                          pjoin(cfg.UPLOAD_FOLDER,
                                "TESTRUN_215153_metadata.dat"),
                          pjoin(cfg.FEATURES_FOLDER,
                                "TEMP_TEST01_featureset.nc"),
                          pjoin(cfg.MODELS_FOLDER, "TEMP_TEST01.pkl"),
                          pjoin(cfg.CUSTOM_FEATURE_SCRIPT_FOLDER,
                                "TESTRUN_CF.py")]:
                try:
                    os.remove(fpath)
                except Exception as e:

After Change

            assert all(key in pred_results_dict for key in \
                       ("ts_data_dict", "features_dict"))
            for fpath in [pjoin(config["paths"]["upload_folder"], "TESTRUN_215153.dat"),
                          pjoin(config["paths"]["upload_folder"],
                                "TESTRUN_215153_metadata.dat"),
                          pjoin(config["paths"]["features_folder"],
                                "TEMP_TEST01_featureset.nc"),
                          pjoin(config["paths"]["models_folder"], "TEMP_TEST01.pkl"),
                          pjoin(config["paths"]["custom_feature_script_folder"],
                                "TESTRUN_CF.py")]:
                try:
                    os.remove(fpath)
                except Exception as e:

Instances