def test_dummy_axis0():
est = Dummy(axis=0)
est.fit(X)
X_predicted = est.predict(X).toarray()
X_expected = [[2, 0, 0, 1], [2, 0, 10. / 3, 0], [0, 4, 10. / 3, 0],
[0, 0, 10. / 3, 0], [2, 0, 0, 0]]
assert_array_almost_equal(X_predicted, X_expected)
def test_dummy_axis1():
est = Dummy()
est.fit(X)
X_predicted = est.predict(X).toarray()
X_expected = [[2, 0, 0, 2], [3.5, 0, 3.5, 0], [0, 3.5, 3.5, 0],
[0, 0, 2, 0], [1, 0, 0, 0]]
assert_array_almost_equal(X_predicted, X_expected)
def test_dummy_axis0():
est = Dummy(axis=0)
est.fit(X)
X_predicted = est.predict(X).toarray()
X_expected = [[2, 0, 0, 1],
[2, 0, 10./3, 0],
[0, 4, 10./3, 0],
[0, 0, 10./3, 0],
[2, 0, 0, 0]]
assert_array_almost_equal(X_predicted, X_expected)
def test_dummy_axis1():
est = Dummy()
est.fit(X)
X_predicted = est.predict(X).toarray()
X_expected = [[2, 0, 0, 2],
[3.5, 0, 3.5, 0],
[0, 3.5, 3.5, 0],
[0, 0, 2, 0],
[1, 0, 0, 0]]
assert_array_almost_equal(X_predicted, X_expected)
