def test_node_response(self):
query = "query"
node = RankNode()
node.load_query(query)
resp = node.response()
stat = [int(word) for word in resp.split() if word.replace('.','').isdigit()]
self.assertTrue(isinstance(resp, str))
def response(self):
if self.node_type == "rank":
node = RankNode(self.query)
elif self.node_type == "stat":
node = RankNode(self.query)
return node.response()
def response(self):
if self.node_type == "rank":
node = RankNode()
elif self.node_type == "stat":
node = StatNode()
elif self.node_type == "info":
node = InfoNode()
node.load_query(self.query)
return node.response()
def test_response(self):
query = "Who is a better shooter Kobe Bryant or Lebron James?"
node = RankNode()
node.load_query(query)
resp = node.response()
self.assertTrue(isinstance(resp, str))