def test_claim(self):
workload_id = uuid.uuid4()
t = Thread(42)
self.assertEqual(None, t.get_workload_id())
t.claim(workload_id)
self.assertEqual(workload_id, t.get_workload_id())
t.free()
self.assertEqual(None, t.get_workload_id())
def test_claim(self):
workload_id = uuid.uuid4()
t = Thread(42)
self.assertEqual(0, len(t.get_workload_ids()))
t.claim(workload_id)
self.assertEqual(1, len(t.get_workload_ids()))
self.assertEqual(workload_id, t.get_workload_ids()[0])
t.clear()
self.assertEqual(0, len(t.get_workload_ids()))
def test_clear_multiple_claims(self):
workload_id_a = "a"
workload_id_b = "b"
t = Thread(42)
t.claim(workload_id_a)
t.claim(workload_id_b)
self.assertTrue(t.is_claimed())
self.assertEqual(2, len(t.get_workload_ids()))
t.clear()
self.assertFalse(t.is_claimed())
self.assertEqual(0, len(t.get_workload_ids()))
def parse_cpu(cpu_dict: dict) -> Cpu:
packages = []
for p in cpu_dict["packages"]:
cores = []
for c in p["cores"]:
threads = []
for t in c["threads"]:
thread = Thread(t["id"])
for w_id in t["workload_id"]:
thread.claim(w_id)
threads.append(thread)
cores.append(Core(c["id"], threads))
packages.append(Package(p["id"], cores))
return Cpu(packages)
def test_free_unknown(self):
t = Thread(42)
self.assertEqual(0, len(t.get_workload_ids()))
self.assertFalse(t.is_claimed())
t.free("unknown_id")
self.assertEqual(0, len(t.get_workload_ids()))
self.assertFalse(t.is_claimed())
workload_id = "a"
t.claim(workload_id)
t.free("unknown_id")
self.assertEqual(1, len(t.get_workload_ids()))
self.assertEqual(workload_id, t.get_workload_ids()[0])
self.assertTrue(t.is_claimed())
def test_get_empty_threads(self):
t0 = Thread(0)
t1 = Thread(1)
t2 = Thread(2)
t3 = Thread(3)
c0 = Core(0, [t0, t2])
c1 = Core(1, [t1, t3])
p = Package(1, [c0, c1])
self.assertEqual([t0, t2, t1, t3], p.get_empty_threads())
t1.claim(uuid.uuid4())
self.assertEqual([t0, t2, t3], p.get_empty_threads())
t1.clear()
self.assertEqual([t0, t2, t1, t3], p.get_empty_threads())
def test_get_emptiest_package(self):
t0 = Thread(0)
t1 = Thread(1)
t2 = Thread(2)
t3 = Thread(3)
t4 = Thread(4)
t5 = Thread(5)
t6 = Thread(6)
t7 = Thread(7)
p0 = Package(0, [Core(0, [t0, t4]), Core(1, [t1, t5])])
p1 = Package(1, [Core(0, [t2, t6]), Core(1, [t3, t7])])
cpu = Cpu([p0, p1])
# The first package should be the emptiest
self.assertEqual(p0, cpu.get_emptiest_package())
# The second package should be the emptiest after we claim a thread on the first
t5.claim(uuid.uuid4())
self.assertEqual(p1, cpu.get_emptiest_package())
# The first package should be the emptiest again, after we release the claimed thread
t5.clear()
self.assertEqual(p0, cpu.get_emptiest_package())
# The first package should be emptiest when we claim a thread on the second
t3.claim(uuid.uuid4())
self.assertEqual(p0, cpu.get_emptiest_package())
# When an equal number of threads are claimed on both packages, the first should be returned
t4.claim(uuid.uuid4())
self.assertEqual(p0, cpu.get_emptiest_package())
def test_equality(self):
t_x = Thread(42)
t_y = Thread(42)
self.assertEqual(t_x, t_y)
t_x.claim("a")
self.assertNotEqual(t_x, t_y)
t_y.claim("a")
self.assertEqual(t_x, t_y)
t_x.claim("b")
t_y.claim("b")
self.assertEqual(t_x, t_y)
def test_multiple_claims(self):
workload_id_a = "a"
workload_id_b = "b"
t = Thread(42)
t.claim(workload_id_a)
self.assertEqual(1, len(t.get_workload_ids()))
self.assertEqual(workload_id_a, t.get_workload_ids()[0])
self.assertTrue(t.is_claimed())
t.claim(workload_id_b)
self.assertEqual(2, len(t.get_workload_ids()))
self.assertTrue(workload_id_a in t.get_workload_ids())
self.assertTrue(workload_id_b in t.get_workload_ids())
self.assertTrue(t.is_claimed())
t.free(workload_id_a)
self.assertEqual(1, len(t.get_workload_ids()))
self.assertEqual(workload_id_b, t.get_workload_ids()[0])
self.assertTrue(t.is_claimed())
t.free(workload_id_b)
self.assertFalse(t.is_claimed())
self.assertEqual(0, len(t.get_workload_ids()))
def test_equality(self):
t_0_0 = Thread(0)
t_0_1 = Thread(1)
c_x = Core(0, [t_0_0, t_0_1])
t_1_0 = Thread(0)
t_1_1 = Thread(1)
c_y = Core(0, [t_1_0, t_1_1])
self.assertEqual(c_x, c_y)
t_0_1.claim("a")
self.assertNotEqual(c_x, c_y)
t_1_1.claim("a")
self.assertEqual(c_x, c_y)
t_0_0.claim("b")
t_1_0.claim("b")
self.assertEqual(c_x, c_y)
def test_equality(self):
t_0_0 = Thread(0)
t_0_1 = Thread(1)
c_x = Core(0, [t_0_0, t_0_1])
p_x = Package(0, [c_x])
cpu_x = Cpu([p_x])
t_1_0 = Thread(0)
t_1_1 = Thread(1)
c_y = Core(0, [t_1_0, t_1_1])
p_y = Package(0, [c_y])
cpu_y = Cpu([p_y])
self.assertEqual(cpu_x, cpu_y)
t_0_1.claim("a")
self.assertNotEqual(cpu_x, cpu_y)
t_1_1.claim("a")
self.assertEqual(cpu_x, cpu_y)
t_0_0.claim("b")
t_1_0.claim("b")
self.assertEqual(cpu_x, cpu_y)