def test_sampleset_id(self):
f = Future(solver=None, id_=None)
# f.id should be None
self.assertIsNone(f.id)
with self.assertRaises(TimeoutError):
f.sampleset.wait_id(timeout=1)
# set it
submission_id = 'test-id'
f.id = submission_id
# validate it's available
self.assertEqual(f.sampleset.wait_id(), submission_id)
self.assertEqual(f.sampleset.wait_id(timeout=1), submission_id)
def test_id_getter_setter(self):
"""Future.get_id/get_id works in isolation as expected."""
f = Future(solver=None, id_=None)
# f.id should be None
self.assertIsNone(f.id)
with self.assertRaises(TimeoutError):
f.wait_id(timeout=1)
# set it
submission_id = 'test-id'
f.id = submission_id
# validate it's available
self.assertEqual(f.wait_id(), submission_id)
self.assertEqual(f.wait_id(timeout=1), submission_id)
self.assertEqual(f.id, submission_id)
