@with_setup(teardown=remove_output)
def test_build_model():
    """Test main model building method - various types"""
    for model_type in MODEL_TYPES:
        if 'Classifier' in model_type:
            fset = xr.open_dataset(pjoin(DATA_PATH, "test_featureset.nc"))
        elif model_type in 'Regressor':
            fset = xr.open_dataset(pjoin(DATA_PATH, "test_reg_featureset.nc"))
        else:
            raise ValueError("Unrecognized scikit-learn model type.")
        model_path = pjoin(TEMP_DIR, "test.pkl")
        build_model.create_and_pickle_model(fset, model_type, model_path)
        model = joblib.load(model_path)
        assert isinstance(model,
                          build_model.MODELS_TYPE_DICT[model_type])
        os.remove(pjoin(TEMP_DIR, "test.pkl"))
dataset/ETHPy150Open cesium-ml/cesium/cesium/tests/test_build_model.py/test_build_model/swapped operands of `in`