def test_idadb_add_column_id_non_destructive_custom_name(self, idadb, idadf_tmp):
idadb.add_column_id(idadf_tmp, "MY_CUSTOM_ID", destructive = False)
assert("MY_CUSTOM_ID" in idadf_tmp.columns)
assert("MY_CUSTOM_ID" == idadf_tmp.indexer)
assert("MY_CUSTOM_ID" not in idadf_tmp._get_all_columns_in_table())
idadf_tmp_new = IdaDataFrame(idadb, idadf_tmp._name)
assert("MY_CUSTOM_ID" not in idadf_tmp_new.columns)
assert("MY_CUSTOM_ID" not in idadf_tmp_new._get_all_columns_in_table())
def test_idadb_delete_column_non_destructive(self, idadb, idadf_tmp):
to_delete = idadf_tmp.columns[0]
idadb.delete_column(idadf_tmp, to_delete, destructive = False)
assert(to_delete not in idadf_tmp.columns)
assert(to_delete in idadf_tmp._get_all_columns_in_table())
idadf_tmp_new = IdaDataFrame(idadb, idadf_tmp._name)
assert(to_delete in idadf_tmp_new.columns)
assert(to_delete in idadf_tmp_new._get_all_columns_in_table())
def test_idadb_delete_column_non_destructive(self, idadb, idadf_tmp):
to_delete = idadf_tmp.columns[0]
idadb.delete_column(idadf_tmp, to_delete, destructive = False)
assert(to_delete not in idadf_tmp.columns)
assert(to_delete in idadf_tmp._get_all_columns_in_table())
idadf_tmp_new = IdaDataFrame(idadb, idadf_tmp._name)
assert(to_delete in idadf_tmp_new.columns)
assert(to_delete in idadf_tmp_new._get_all_columns_in_table())
def test_idadb_add_column_id_non_destructive_custom_name(self, idadb, idadf_tmp):
idadb.add_column_id(idadf_tmp, "MY_CUSTOM_ID", destructive = False)
assert("MY_CUSTOM_ID" in idadf_tmp.columns)
assert("MY_CUSTOM_ID" == idadf_tmp.indexer)
assert("MY_CUSTOM_ID" not in idadf_tmp._get_all_columns_in_table())
idadf_tmp_new = IdaDataFrame(idadb, idadf_tmp._name)
assert("MY_CUSTOM_ID" not in idadf_tmp_new.columns)
assert("MY_CUSTOM_ID" not in idadf_tmp_new._get_all_columns_in_table())
def test_idadb_delete_column_non_destructive_virtual_indexer(self, idadb, idadf_tmp):
idadb.add_column_id(idadf_tmp, "ID", destructive = False)
idadb.delete_column(idadf_tmp, "ID", destructive = False)
assert("ID" not in idadf_tmp.columns)
assert("ID" != idadf_tmp.indexer)
assert("ID" not in idadf_tmp._get_all_columns_in_table())
idadf_tmp_new = IdaDataFrame(idadb, idadf_tmp._name)
assert("ID" not in idadf_tmp_new.columns)
assert("ID" not in idadf_tmp_new._get_all_columns_in_table())
def test_idadb_add_column_id_destructive(self, idadb, idadf_tmp):
idadb.add_column_id(idadf_tmp, "ID", destructive = True)
assert("ID" in idadf_tmp.columns)
assert("ID" == idadf_tmp.indexer)
assert("ID" in idadf_tmp._get_all_columns_in_table())
idadf_tmp_new = IdaDataFrame(idadb, idadf_tmp._name, indexer = "ID")
assert("ID" in idadf_tmp_new.columns)
assert("ID" == idadf_tmp_new.indexer)
assert("ID" in idadf_tmp_new._get_all_columns_in_table())
def test_idadb_delete_column_non_destructive_virtual_indexer(self, idadb, idadf_tmp):
idadb.add_column_id(idadf_tmp, "ID", destructive = False)
idadb.delete_column(idadf_tmp, "ID", destructive = False)
assert("ID" not in idadf_tmp.columns)
assert("ID" != idadf_tmp.indexer)
assert("ID" not in idadf_tmp._get_all_columns_in_table())
idadf_tmp_new = IdaDataFrame(idadb, idadf_tmp._name)
assert("ID" not in idadf_tmp_new.columns)
assert("ID" not in idadf_tmp_new._get_all_columns_in_table())
def test_idadb_add_column_id_destructive(self, idadb, idadf_tmp):
idadb.add_column_id(idadf_tmp, "ID", destructive = True)
assert("ID" in idadf_tmp.columns)
assert("ID" == idadf_tmp.indexer)
assert("ID" in idadf_tmp._get_all_columns_in_table())
idadf_tmp_new = IdaDataFrame(idadb, idadf_tmp._name, indexer = "ID")
assert("ID" in idadf_tmp_new.columns)
assert("ID" == idadf_tmp_new.indexer)
assert("ID" in idadf_tmp_new._get_all_columns_in_table())