class Test(unittest.TestCase): def setUp(self): class Stub(object): obj_grads = [10,0] def add_observer(self, m, f): self.obs = m self.obs_f = f def objective_function(self): return 10 self.vo = VerboseOptimization(Stub(), opt_bfgs(), -10, verbose=True) def test_timestrings(self): self.vo.print_out(0) self.assertEqual(self.vo.timestring, '00s00') self.vo.print_out(10.2455) self.assertEqual(self.vo.timestring, '10s24') self.vo.print_out(120) self.assertEqual(self.vo.timestring, '02m00s00') self.vo.print_out(60*60+120+12.2455) self.assertEqual(self.vo.timestring, '01h02m12') self.vo.print_out(2*3600*24+60*60+120+12.2455) self.assertEqual(self.vo.timestring, '02d01h02') def test_finish(self): self.assertEqual(self.vo.status, 'running')
def setUp(self): class Stub(object): obj_grads = [10,0] def add_observer(self, m, f): self.obs = m self.obs_f = f def objective_function(self): return 10 self.vo = VerboseOptimization(Stub(), opt_bfgs(), -10, verbose=True)