def test_traverse_project_when_not_marked (self): parent = Project (name=u'p') n1 = Task (name=u'n1') n2 = Task (name=u'n2') parent.children.append (n1) parent.children.append (n2) parent.marked = False visitor = DemoVisitor () traverse (visitor, parent) self.assertEqual(0, len(visitor.tasks_started)) self.assertEqual(0, len(visitor.projects_started)) self.assertEqual(0, len(visitor.tasks_ended)) self.assertEqual(0, len(visitor.projects_ended))
def test_traverse_project_when_not_marked(self): parent = Project(name=u'p') n1 = Task(name=u'n1') n2 = Task(name=u'n2') parent.children.append(n1) parent.children.append(n2) parent.marked = False visitor = DemoVisitor() traverse(visitor, parent) self.assertEqual(0, len(visitor.tasks_started)) self.assertEqual(0, len(visitor.projects_started)) self.assertEqual(0, len(visitor.tasks_ended)) self.assertEqual(0, len(visitor.projects_ended))