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)