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