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()