

def test_issue_341():
    data = sm.datasets.anes96.load()
    exog = data.exog
    exog = exog[:, :(- 1)]
    exog = sm.add_constant(exog, prepend=True)
    res1 = sm.MNLogit(data.endog, exog).fit(method='newton', disp=0)
    x = exog[0]
    np.testing.assert_equal(res1.predict(x).shape, (1, 7))
    np.testing.assert_equal(x.predict(x[None]).shape, (1, 7))

dataset/ETHPy150Open statsmodels/statsmodels/statsmodels/discrete/tests/test_discrete.py test_issue_341/VarMisuse@112/134 `res1`->`x`