def test_hybrid(self): df_min = gr.eval_hybrid(self.md, df_det="nom") self.assertTrue( set(df_min.columns) == set(self.md.var + self.md.out + ["hybrid_var"])) self.assertTrue(df_min._meta["type"] == "eval_hybrid") df_seeded = gr.eval_hybrid(self.md, df_det="nom", seed=101) df_piped = self.md >> gr.ev_hybrid(df_det="nom", seed=101) self.assertTrue(df_seeded.equals(df_piped)) df_total = gr.eval_hybrid(self.md, df_det="nom", plan="total") self.assertTrue( set(df_total.columns) == set(self.md.var + self.md.out + ["hybrid_var"])) self.assertTrue(df_total._meta["type"] == "eval_hybrid") df_skip = gr.eval_hybrid(self.md, df_det="nom", skip=True) self.assertTrue( set(df_skip.columns) == set(self.md.var + ["hybrid_var"])) ## Raises md_buckle = models.make_plate_buckle() with self.assertRaises(ValueError): gr.eval_hybrid(md_buckle, df_det="nom")
def test_pipe(self): ## Chain res = self.md >> gr.ev_hybrid(df_det="nom") >> gr.tf_sobol()