def test_get(self):
p = sp.SpearmintProposer(self.pc)
for j in range(3):
tid = []
for i in range(10):
v = p.get()
tid.append(v['job_id'])
for i in tid:
job = Job("none", {'job_id': i})
p.update(1, job)
self.assertTrue(True)
self.assertTrue(sp.SpearmintProposer(self.pc))
def test_init(self):
pc = self.pc.copy()
pc['engine'] = "wrong"
self.assertRaises(KeyError, sp.SpearmintProposer, pc)
pc = self.pc.copy()
pc['parameter_config'][0]['size'] = 2
p = sp.SpearmintProposer(pc)
self.assertRaises(NotImplementedError, p.get)
pc = self.pc.copy()
pc['parameter_config'][0]['name'] = 'job_id'
self.assertRaises(ValueError, sp.SpearmintProposer, pc)
def test_finish(self):
p = sp.SpearmintProposer(self.pc)
self.assertRaises(NotImplementedError, p.save, "")
self.assertRaises(NotImplementedError, p.reload, "")
def test_failed(self):
p = sp.SpearmintProposer(self.pc)
c = p.get()
job = Job("none", c)
p.failed(job)