Python：在测试数据中检测EOF

如何解决Python：在测试数据中检测EOF

我在通过python自动测试时遇到了问题。而且我认为原因是我的程序在测试数据结束时没有结束。

该程序使用以下命令在两个IP地址之间建立网络：

• B 100.100.100.1 100.100.100.2

这意味着：以两种方式在IP1和IP2之间建立连接。

还有第二个命令来测试连接是否存在。

• T 100.100.100.1 100.100.100.3

如果可能，程序将打印“ T”，否则显示“ N”。

让我们说输入数据如下：

B 100.100.100.1 100.100.100.2
B 100.100.100.1 100.100.100.3
B 100.100.100.10 100.100.100.11
T 100.100.100.1 100.100.100.3
T 100.100.100.10 100.100.100.2
T 100.100.100.10 100.100.100.11
B 100.100.100.11 100.100.100.2
T 100.100.100.10 100.100.100.3
T 100.100.100.100 100.100.100.103

则预期输出为：

T
N
T
T
N

我的程序打印出正确的值，但是如果测试数据结束，则不会出错，并且一直工作到检测到输入中的空行为止。但这需要在命令行中再按1个“ Enter”键才能发送此行，并且自动测试不会发送它。

我认为我需要一次读取所有输入数据，但是我不知道如何在每一行的末尾忽略'\ n'，而在输入末尾仅检测到1个“ enter”。

我希望问题描述清楚。这是代码：

from collections import defaultdict
class Graph:

    def __init__(self,vertices):
        self.V= vertices
        self.graph = defaultdict(list)
    def addVerticle(self):
        self.V+=1

    def addEdge(self,u,v):
        self.graph[u].append(v)

    def graph_keys(self):
        return self.graph.keys()

    def isReachable(self,s,d):
        visited =[False]*(self.V)
        queue=[]
        queue.append(s)
        visited[s] = True

        while queue:
            n = queue.pop(0)
            if n == d:
                 return True
            for i in self.graph[n]:
                if visited[i] == False:
                    queue.append(i)
                    visited[i] = True
        return False

ip_list={}
counter=0
g = Graph(0)
while True:
    u_in = [x for x in input().split()]
    if not u_in:
        break
    action = u_in[0]
    start = u_in[1]
    end = u_in[2]
    if action == 'B':
        if start not in ip_list.keys():
            ip_list[start]=counter
            counter+=1
        if end not in ip_list.keys():
            ip_list[end]=counter
            counter+=1
        start = ip_list[start]
        end = ip_list[end]
        if start not in g.graph_keys():
            g.addVerticle()
        if end not in g.graph_keys():
            g.addVerticle()
        g.addEdge(start,end)
        g.addEdge(end,start)
    elif action =='T':
        if start not in ip_list.keys() or end not in ip_list.keys():
            print('N')
            continue
        if g.isReachable(ip_list[start],ip_list[end]):
            print('T')
        else:
            print('N')

解决方法

下面（假设输入是来自文件的comimg）

with open('input.txt') as f:
  lines = [l.strip() for l in f.readlines()]
for line in lines:
  print(line)
  # TODO - your logic goes here

input.txt

B 100.100.100.1 100.100.100.2
B 100.100.100.1 100.100.100.3
B 100.100.100.10 100.100.100.11
T 100.100.100.1 100.100.100.3
T 100.100.100.10 100.100.100.2
T 100.100.100.10 100.100.100.11
B 100.100.100.11 100.100.100.2
T 100.100.100.10 100.100.100.3
T 100.100.100.100 100.100.100.103