Python coerce_queries_dataframe示例

示例#1

文件： test_model.py 项目： marcosfpr/pyterrier

 def test_coerce_dataframe_with_tuple(self):
     input = ("light", "mathematical", "electronic")
     exp_result = pd.DataFrame(
         [["1", "light"], ["2", "mathematical"], ["3", "electronic"]],
         columns=['qid', 'query'])
     result = coerce_queries_dataframe(input)
     self.assertTrue(exp_result.equals(result))

示例#2

 def test_coerce_dataframe_with_string(self):
     input = "light"
     exp_result = pd.DataFrame([["1", "light"]], columns=['qid', 'query'])
     result = coerce_queries_dataframe(input)
     self.assertTrue(exp_result.equals(result))