def test_binary_blocks():
X, Y = toy.generate_blocks(n_samples=10)
crf = GridCRF()
clf = StructuredPerceptron(problem=crf, max_iter=30)
clf.fit(X, Y)
Y_pred = clf.predict(X)
assert_array_equal(Y, Y_pred)
def test_multinomial_blocks():
X, Y = toy.generate_blocks_multinomial(n_samples=10, noise=0.3, seed=0)
crf = GridCRF(n_states=X.shape[-1])
clf = StructuredPerceptron(problem=crf, max_iter=10)
clf.fit(X, Y)
Y_pred = clf.predict(X)
assert_array_equal(Y, Y_pred)
