def test_extract_metric(self): query = "Who is a better shooter Kobe or Lebron?" node = RankNode() node.load_query(query) metric = node.extract_metric() self.assertEqual(metric, "shooter")
def test_extract_metric(self): node = RankNode("Who is a better shooter Kobe or Lebron?") metric = node.extract_metric() self.assertEqual(metric, "shooter")