def testSetGetStarts(self, procs, starts): for p in procs: ProcTree.register(p) pt = ProcTree() pt.setStarts(starts) for proc in procs: if proc in starts: self.assertTrue(ProcTree.NODES[proc].start) else: self.assertFalse(ProcTree.NODES[proc].start) s = pt.getStarts() self.assertCountEqual(s, starts) self.assertCountEqual(pt.starts, starts)
def test_proctree_setgetstarts(): p12 = Proc() p13 = Proc() p14 = Proc() #p14.hide = True p12.depends = p13 ProcTree.register(p13) ProcTree.register(p14) pt = ProcTree() pt.init() # with pytest.raises(ProcHideError): # pt.setStarts([p14]) pt.ends = [1, 2, 3] pt.setStarts([p13]) assert ProcTree.NODES[p12].start == False assert ProcTree.NODES[p13].start == True assert ProcTree.NODES[p14].start == False assert pt.starts == [p13] assert pt.ends == [] pt.starts = [] assert pt.getStarts() == [p13] assert pt.starts == [p13] assert pt.getStarts() == [p13]