09bfc1b3ef8494cf17a192bbe83df576ccdac86f,cesium/tests/test_predict.py,,test_predict_classification,#,34

Before Change

                                                       model,
                                                       custom_features_script=None)
        for fname, results in pred_results_dict.items():
            for el in results["pred_results"]:
                assert(el[0] in [b"class1", b"class2", b"class3"]
                       or el in [b"class1", b"class2", b"class3"])


def test_predict_regression():
    Test main predict function on multiple files (regression)
    regressor_types = [model_type for model_type, model_class
                       in build_model.MODELS_TYPE_DICT.items()

After Change

        preds = predict.predict_data_files(TS_CLASS_PATHS,
                                           list(fset.data_vars), model,
                                           custom_features_script=None)
        if preds.prediction.values.ravel()[0].dtype == np.dtype("float"):
            assert(all(preds.prediction.class_label == [b"class1", b"class2",
                                                        b"class3"]))
            assert(preds.prediction.values.shape ==

Instances