3 from gackshandle import *
5 class TestGacksHandle(unittest.TestCase):
10 h = GacksHandle("cpu", 10, 15, 20, 25)
11 self.assertEqual(h.id, "cpu")
12 self.assertEqual(h.unitStart, 10)
13 self.assertEqual(h.unitStop, 15)
14 self.assertEqual(h.timeStart, 20)
15 self.assertEqual(h.timeStop, 25)
17 def testAsString(self):
18 h = GacksHandle("cpu", 10, 15, 20, 25)
20 self.assertEqual(s, "cpu#10-15#20-25")
22 def testLoadFromString(self):
23 h = GacksHandle("cpu", 10, 15, 20, 25)
25 h2 = GacksHandle(string=s)
26 self.assertEqual(h.id, h2.id)
27 self.assertEqual(h.unitStart, h2.unitStart)
28 self.assertEqual(h.unitStop, h2.unitStop)
29 self.assertEqual(h.timeStart, h2.timeStart)
30 self.assertEqual(h.timeStop, h2.timeStop)
32 def testGetQuantity(self):
33 h = GacksHandle("cpu", 10, 15, 20, 25)
34 self.assertEqual(h.get_quantity(), 5)
36 def testGetDuration(self):
37 h = GacksHandle("cpu", 10, 15, 20, 25)
38 self.assertEqual(h.get_duration(), 5)
41 h = GacksHandle("cpu", 10, 15, 20, 25)
43 self.assertEqual(h.id, h2.id)
44 self.assertEqual(h.unitStart, h2.unitStart)
45 self.assertEqual(h.unitStop, h2.unitStop)
46 self.assertEqual(h.timeStart, h2.timeStart)
47 self.assertEqual(h.timeStop, h2.timeStop)
50 h = GacksHandle("cpu", 10, 15, 20, 25)
51 (h1, h2) = h.split(12,23)
53 self.assertEqual(h1.id, "cpu")
54 self.assertEqual(h1.unitStart, 10)
55 self.assertEqual(h1.unitStop, 12)
56 self.assertEqual(h1.timeStart, 20)
57 self.assertEqual(h1.timeStop, 23)
59 self.assertEqual(h2.id, "cpu")
60 self.assertEqual(h2.unitStart, 12)
61 self.assertEqual(h2.unitStop, 15)
62 self.assertEqual(h2.timeStart, 23)
63 self.assertEqual(h2.timeStop, 25)
65 class TestGacksRecord(unittest.TestCase):
70 r = GacksRecord("cpu", 10, 15, 20, 25, ["foo","bar"], "slice1")
71 self.assertEqual(r.id, "cpu")
72 self.assertEqual(r.unitStart, 10)
73 self.assertEqual(r.unitStop, 15)
74 self.assertEqual(r.timeStart, 20)
75 self.assertEqual(r.timeStop, 25)
76 self.assertEqual(r.allocatorHRNs, ["foo", "bar"])
77 self.assertEqual(r.consumerHRN, "slice1")
80 r = GacksRecord("cpu", 10, 15, 20, 25, ["foo","bar"], "slice1")
83 self.assertEqual(r.id, r2.id)
84 self.assertEqual(r.unitStart, r2.unitStart)
85 self.assertEqual(r.unitStop, r2.unitStop)
86 self.assertEqual(r.timeStart, r2.timeStart)
87 self.assertEqual(r.timeStop, r2.timeStop)
88 self.assertEqual(r.allocatorHRNs, r2.allocatorHRNs)
89 self.assertEqual(r.consumerHRN, r2.consumerHRN)
91 def testSetAllocator(self):
92 r = GacksRecord("cpu", 10, 15, 20, 25, ["foo","bar"], "slice1")
93 r.set_allocator("bar", "bob", -1, 1)
94 self.assertEqual(r.allocatorHRNs, ["foo", "bar", "bob"])
96 r = GacksRecord("cpu", 10, 15, 20, 25, ["foo","bar"], "slice1")
97 r.set_allocator("bar", "bob", -1, 0)
98 self.assertEqual(r.allocatorHRNs, ["foo", "bob"])
100 r = GacksRecord("cpu", 10, 15, 20, 25, ["foo","bar","foo","bar"], "slice1")
101 r.set_allocator("bar", "bob", 0, 0)
102 self.assertEqual(r.allocatorHRNs, ["foo", "bob"])
104 r = GacksRecord("cpu", 10, 15, 20, 25, ["foo","bar","foo","bar"], "slice1")
105 r.set_allocator("bar", "bob", 1, 0)
106 self.assertEqual(r.allocatorHRNs, ["foo", "bar", "foo", "bob"])
108 def testGetAllocator(self):
109 r = GacksRecord("cpu", 10, 15, 20, 25, ["foo","bar"], "slice1")
110 self.assertEqual(r.get_allocators(), ["foo", "bar"])
112 def testSetConsumer(self):
113 r = GacksRecord("cpu", 10, 15, 20, 25, ["foo","bar"], "slice1")
114 r.set_consumer("slice2")
115 self.assertEqual(r.get_consumer(), "slice2")
117 def testGetConsumer(self):
118 r = GacksRecord("cpu", 10, 15, 20, 25, ["foo","bar"], "slice1")
119 self.assertEqual(r.get_consumer(), "slice1")
123 if __name__ == "__main__":