def test_task_order(sentence1): node_0 = Node('', 0, sentence1, state=NodeState.ROOT) tasklist = TaskList(node_0) node_1a = Node('6', 6, '', parent=node_0) tasklist.add(node_1a) node_1b = Node('5', 5, '', parent=node_0) tasklist.add(node_1b) node_1c = Node('2', 2, '', parent=node_0) tasklist.add(node_1c) node_1d = Node('3', 3, '', parent=node_0) tasklist.add(node_1d) node_1e = Node('4', 4, '', parent=node_0) tasklist.add(node_1e) node_1f = Node('1', 1, '', parent=node_0) tasklist.add(node_1f) assert len(tasklist) == 7 next_node = tasklist.next() assert next_node.next_pos == 7 assert len(tasklist) == 6 next_node = tasklist.next() assert next_node.next_pos == 6 assert len(tasklist) == 5
def test_take_2(sentence1): node_0 = Node('', 0, sentence1, state=NodeState.ROOT) tasklist = TaskList(node_0) assert len(tasklist) == 1 nodes = tasklist.next(size=2) assert len(nodes) == 1 next_node_0 = nodes[0] assert len(tasklist) == 0 assert next_node_0.next_pos == 0 node_1 = Node('..', 0, '-.--.', parent=next_node_0) tasklist.add(node_1) node_2 = Node('-', 2, '.--.', parent=node_1) tasklist.add(node_2) nodes = tasklist.next(size=2) assert len(nodes) == 2
def test_simple_path(sentence1): node_0 = Node('', 0, sentence1, state=NodeState.ROOT) tasklist = TaskList(node_0) assert len(tasklist) == 1 next_node_0 = tasklist.next() assert len(tasklist) == 0 assert next_node_0.next_pos == 0 node_1 = Node('..', 0, '-.--.', parent=next_node_0) tasklist.add(node_1) next_node_1 = tasklist.next() assert len(tasklist) == 0 assert next_node_1.next_pos == 2 node_2 = Node('-', 2, '.--.', parent=next_node_1) tasklist.add(node_2) next_node_2 = tasklist.next() assert len(tasklist) == 0 assert next_node_2.next_pos == 3 node_3 = Node('.--', 3, '.', parent=next_node_2) tasklist.add(node_3) next_node_3 = tasklist.next() assert len(tasklist) == 0 assert next_node_3.next_pos == 6 node_4 = Node('.', 6, '', parent=next_node_3) tasklist.add(node_4) next_node_4 = tasklist.next() assert len(tasklist) == 0 assert next_node_4.next_pos == 7 next_node_done = tasklist.next() assert next_node_done is None
def test_first_task(sentence1): node_0 = Node('', 0, sentence1, state=NodeState.ROOT) tasklist = TaskList(node_0) next_node = tasklist.next() assert next_node.remaining == sentence1