def setUp(self): self.pcb_table = PCBTable() self.pcb1 = PCB(3, 5, 20, 40, 4) self.pcb2 = PCB(8, 10, 25, 42, 3) self.pcb3 = PCB(20, 30, 15, 45, 5) self.pcb_table.add(self.pcb1) self.pcb_table.add(self.pcb2) self.pcb_table.add(self.pcb3)
class TestPCBTable(unittest.TestCase): def setUp(self): self.pcb_table = PCBTable() self.pcb1 = PCB(3, 5, 20, 40, 4) self.pcb2 = PCB(8, 10, 25, 42, 3) self.pcb3 = PCB(20, 30, 15, 45, 5) self.pcb_table.add(self.pcb1) self.pcb_table.add(self.pcb2) self.pcb_table.add(self.pcb3) def test_add(self): self.assertEqual(3, self.pcb_table.size()) def test_remove(self): self.pcb_table.remove(self.pcb2) self.assertEqual(2, self.pcb_table.size())