def METHOD_NAME(self):
    model = gramex.ml.load(self.model_path)
    result = model.predict(
        [
            {
                'sepal_length': 5.7,
                'sepal_width': 4.4,
                'petal_length': 1.5,
                'petal_width': 0.4,
            }
        ]
    )
    eq_(result.tolist(), ['setosa'])