def createContJob():
client = TroiaContClient(ADDRESS)
client.create()
jobs.append((client.jid, "CONTINUOUS"))
check_status(client, client.post_assigned_labels(ASSIGNED_LABELS_CONT))
check_status(client, client.post_gold_data(GOLD_LABELS_CONT))
check_assigns(client, len(ASSIGNED_LABELS_CONT))
class ContSimulation(Simulation):
def create(self):
self.tc = TroiaContClient(TROIA_ADDRESS)
self.tc.create()
def _upload_golds(self, golds):
alt_golds = ((obj, val, zeta) for obj, (val, zeta) in golds)
return ret_exectime(self.tc, self.tc.post_gold_data(alt_golds))
def _upload_assigns_package(self, assigns):
return ret_exectime(self.tc, self.tc.post_assigned_labels(assigns))
def compute(self):
return ret_exectime(self.tc, self.tc.post_compute())
def shutdown(self):
self.tc.delete()
class TestPrediction(unittest.TestCase):
def setUp(self):
self.client = TroiaContClient(ADDRESS)
response = self.client.create(iterations=20)
self.assertEqual('OK', response['status'])
#post the assigned labels
response = self.client.await_completion(self.client.post_assigned_labels(ASSIGNED_LABELS_CONT))
self.assertEqual('OK', response['status'])
#post golds
response = self.client.await_completion(self.client.post_gold_data(GOLD_LABELS_CONT))
self.assertEqual('OK', response['status'])
def tearDown(self):
self.client.delete()
def test_Compute(self):
response = self.client.await_completion(self.client.post_compute())
self.assertEqual('OK', response['status'])
def test_GetPredictionObjects_WithCompute(self):
response = self.client.await_completion(self.client.post_compute())
self.assertEqual('OK', response['status'])
response = self.client.await_completion(self.client.get_objects_prediction())
self.assertEqual('OK', response['status'])
self.assertEqual(len(EXPECTED_PREDICTION_OBJECTS), len(response['result']))
for object in response['result']:
object_name = object['object']
self.assertAlmostEqual(EXPECTED_PREDICTION_OBJECTS[object_name][0], object['prediction']['est_zeta'], 5)
def test_GetPredictionForOneObject_WithCompute(self):
response = self.client.await_completion(self.client.post_compute())
self.assertEqual('OK', response['status'])
response = self.client.await_completion(self.client.get_object_prediction('url1'))
self.assertEqual('OK', response['status'])
self.assertNotEqual({}, response['result'])
def test_GetPredictionWorkers_WithCompute(self):
response = self.client.await_completion(self.client.post_compute())
self.assertEqual('OK', response['status'])
response = self.client.await_completion(self.client.get_workers_prediction())
self.assertEqual('OK', response['status'])
result = response['result']
self.assertEqual(5, len(result))
for worker in result:
#check the assigned labels
worker_name = worker['worker']
self.assertAlmostEqual(EXPECTED_PREDICTION_WORKERS[worker_name][0], worker['quality']['est_mu'], 5)
self.assertAlmostEqual(EXPECTED_PREDICTION_WORKERS[worker_name][1], worker['quality']['est_sigma'], 5)
self.assertAlmostEqual(EXPECTED_PREDICTION_WORKERS[worker_name][2], worker['quality']['est_rho'], 5)
def test_GetPredictionForOneWorker_WithCompute(self):
response = self.client.await_completion(self.client.post_compute())
self.assertEqual('OK', response['status'])
response = self.client.await_completion(self.client.get_worker_prediction('worker1'))
self.assertEqual('OK', response['status'])
self.assertNotEqual({}, response['result'])
