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)