コード例 #1
0
    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")
コード例 #2
0
ファイル: test_models.py プロジェクト: rfenrich/py_grama
    def test_make(self):
        ## Models build
        md_cantilever_beam = models.make_cantilever_beam()
        md_ishigami = models.make_ishigami()
        md_linear_normal = models.make_linear_normal()
        md_plane_laminate = models.make_composite_plate_tension([0])
        md_plate_buckling = models.make_plate_buckle()
        md_poly = models.make_poly()
        md_test = models.make_test()
        md_trajectory_linear = models.make_trajectory_linear()

        ## Models evaluate
        df_cantilever = md_cantilever_beam >> gr.ev_nominal(df_det="nom")
        df_ishigami = md_ishigami >> gr.ev_nominal(df_det="nom")
        df_ln = md_linear_normal >> gr.ev_nominal(df_det="nom")
        df_plane = md_plane_laminate >> gr.ev_nominal(df_det="nom")
        df_plate = md_plate_buckling >> gr.ev_nominal(df_det="nom")
        df_poly = md_poly >> gr.ev_nominal(df_det="nom")
        df_test = md_test >> gr.ev_nominal(df_det="nom")
        df_traj = md_trajectory_linear >> gr.ev_nominal(df_det="nom")