def test_vectorize_sklearn(constraint, axis):
# get dataset
from sklearn.datasets import load_iris
iris = load_iris()
# build transform
ineq = vectorize(constraint, axis)
from sklearn.preprocessing import FunctionTransformer
t = FunctionTransformer(func=ineq, validate=False) #XXX: inverse?
# test transform
import numpy as np
iris_ = t.fit(iris.data).transform(iris.data)
assert np.all(t._transform(iris_) == iris_)
