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))