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

Before Change

                                                       list(fset.data_vars),
                                                       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

            assert(preds.prediction.values.shape ==
                   (len(TS_CLASS_PATHS), len(np.unique(fset.target))))
        else:
            assert(all(p in [b"class1", b"class2", b"class3"] for p in
                       preds.prediction))


def test_predict_regression():
    Test main predict function on multiple files (regression)

Instances