def test_insert_static_data_4(self): query_string = "item1 item2 item3" query = TokenQuery(query_string, self.session) query._query_id = 999 options.cfg.selected_features = ["coquery_query_token"] df = query.insert_static_data(self.df) self.assertListEqual( sorted(df.columns.tolist()), sorted([ "coq_word_label_1", "coq_word_label_2", "coq_word_label_3", "coquery_invisible_corpus_id", "coquery_dummy", "coquery_invisible_query_id", "coquery_query_token_1", "coquery_query_token_2", "coquery_query_token_3" ])) self.assertListEqual(df.coquery_query_token_1.tolist(), ["item1"] * len(df)) self.assertListEqual(df.coquery_query_token_2.tolist(), ["item2"] * len(df)) self.assertListEqual(df.coquery_query_token_3.tolist(), ["item3"] * len(df))
def test_insert_static_data_2(self): query = TokenQuery("item1{0,3} item2{1,3} item3{2,3}", self.session) query._query_id = 999 df = query.insert_static_data(self.df) self.assertListEqual(df.columns.tolist(), [ "coq_word_label_1", "coq_word_label_2", "coq_word_label_3", "coquery_invisible_corpus_id", "coquery_dummy", "coquery_invisible_query_id" ])
def test_insert_static_data_2(self): query = TokenQuery("item1{0,3} item2{1,3} item3{2,3}", self.session) query._query_id = 999 df = query.insert_static_data(self.df) self.assertListEqual( df.columns.tolist(), ["coq_word_label_1", "coq_word_label_2", "coq_word_label_3", "coquery_invisible_corpus_id", "coquery_dummy", "coquery_invisible_query_id"])
def test_insert_static_data_3(self): query_string = "item1 item2 item3" query = TokenQuery(query_string, self.session) query._query_id = 999 options.cfg.selected_features = ["word_label", "coquery_query_string"] df = query.insert_static_data(self.df) self.assertListEqual( sorted(df.columns.tolist()), sorted(["coq_word_label_1", "coq_word_label_2", "coq_word_label_3", "coquery_invisible_corpus_id", "coquery_dummy", "coquery_invisible_query_id", "coquery_query_string"])) self.assertListEqual( df.coquery_query_string.tolist(), [query_string] * len(df))