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