def test_predict_insert_into_df(self):
ts = 0
te = self.n - 1
modobj = SMMNLogit(self.model_description)
modobj.train(self.df)
modobj.populate(self.nsim)
prediction, varname = modobj.predict(sim=0, data=self.df.loc[ts:te])
self.df[varname] = 0
self.df.loc[ts:te, varname] = prediction[0]
def test_predict_returns_tuple(self):
modobj = SMMNLogit(self.model_description)
modobj.train(self.df)
modobj.populate(self.nsim)
prediction = modobj.predict(sim=0, data=self.df.loc[5])
self.assertIsInstance(prediction, tuple)