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)