def test_multiple_worker_no_matrices(self): worker_1 = Worker("foo", "lo") worker_2 = Worker("foo", "lo") workers = {"worker_1": worker_1, "worker_2": worker_2} result = ortools_interface.allocate_services_to_workers( workers=workers, services=["service_1"]) self.assertIsNone(result)
def test_multiple_workers_zero_services(self): workers = { "worker_1": Worker("foo", "lo"), "worker_2": Worker("foo", "lo") } result = ortools_interface.allocate_services_to_workers( workers=workers) self.assertIsNone(result)
def test_mismatched_matrices(self): worker_1 = Worker("foo", "lo") worker_2 = Worker("foo", "lo") workers = {"worker_1": worker_1, "worker_2": worker_2} result = ortools_interface.allocate_services_to_workers( workers=workers, services=["service_1", "service_2"], hardware_matrix=[[1], [1, 1]], cost_matrix=[[4]], capacity_matrix=[[1], [1]]) self.assertIsNone(result)
def test_multiple_worker_empty_matrices(self): worker_1 = Worker("foo", "lo") worker_2 = Worker("foo", "lo") workers = {"worker_1": worker_1, "worker_2": worker_2} result = ortools_interface.allocate_services_to_workers( workers=workers, services=["service_1"], hardware_matrix=[], cost_matrix=[], capacity_matrix=[]) self.assertIsNone(result)
def test_autogenerated_dependency_matrix(self): worker_1 = Worker("foo", "lo") worker_2 = Worker("foo", "lo") workers = {"worker_1": worker_1, "worker_2": worker_2} result = ortools_interface.allocate_services_to_workers( workers=workers, services=["service_1", "service_2"], hardware_matrix=[[1, 1], [1, 1]], cost_matrix=[[4, 3], [4, 3]]) self.assertTrue((result[worker_1] == ["service_1"] and result[worker_2] == ["service_2"]) or (result[worker_1] == ["service_2"] and result[worker_2] == ["service_1"]))
def test_one_worker_multiple_services(self): worker_1 = Worker("foo", "lo") workers = {"worker_1": worker_1} result = ortools_interface.allocate_services_to_workers( workers=workers, services=["service_1", "service_2"]) self.assertEqual({worker_1: ["service_1", "service_2"]}, result)