Ejemplo n.º 1
0
    def testEq(self):
        new_batch_trial = self.experiment.new_batch_trial()
        self.assertNotEqual(self.batch, new_batch_trial)

        abandoned_arm = get_abandoned_arm()
        abandoned_arm_2 = get_abandoned_arm()
        self.assertEqual(abandoned_arm, abandoned_arm_2)
Ejemplo n.º 2
0
    def testSortable(self):
        new_batch_trial = self.experiment.new_batch_trial()
        self.assertTrue(self.batch < new_batch_trial)

        abandoned_arm = get_abandoned_arm()
        abandoned_arm_2 = get_abandoned_arm()
        abandoned_arm_2.name = "0_1"
        self.assertTrue(abandoned_arm < abandoned_arm_2)

        generator_run = get_generator_run()
        generator_run_struct = GeneratorRunStruct(generator_run=generator_run,
                                                  weight=1.0)
        generator_run_struct_2 = GeneratorRunStruct(
            generator_run=generator_run, weight=2.0)
        self.assertTrue(generator_run_struct < generator_run_struct_2)