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