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())
def test_idadb_as_idadataframe_same_tablename(self, idadb, df):
ida11 = idadb.as_idadataframe(df,
"TEST_AS_IDADF_18729493954_23849590",
clear_existing=True)
ida21 = idadb.as_idadataframe(
df,
"DUMMY.TEST_AS_IDADF_18729493954_23849590",
clear_existing=True)
assert (all(ida11.columns == df.columns))
assert (all(ida21.columns == df.columns))
ida12 = IdaDataFrame(idadb, "TEST_AS_IDADF_18729493954_23849590")
ida22 = IdaDataFrame(idadb, "DUMMY.TEST_AS_IDADF_18729493954_23849590")
assert (all(ida12.columns == df.columns))
assert (all(ida22.columns == df.columns))
idadb.drop_table("TEST_AS_IDADF_18729493954_23849590")
idadb.drop_table("DUMMY.TEST_AS_IDADF_18729493954_23849590")
def test_idadb_create_table(self, idadb, df):
try:
idadb.drop_table("CREATE_TABLE_TEST_585960708904")
except:
pass
idadb._create_table(df, "CREATE_TABLE_TEST_585960708904")
ida = IdaDataFrame(idadb, "CREATE_TABLE_TEST_585960708904")
assert (all(
[str(x) == str(y) for x, y in zip(df.columns, ida.columns)]))
assert (idadb.exists_table("CREATE_TABLE_TEST_585960708904") == 1)
idadb.drop_table("CREATE_TABLE_TEST_585960708904")
def test_idadb_insert_into_database(self, idadb, df):
try:
idadb.drop_table("INSERT_TEST_585960708904")
except:
pass
idadb._create_table(df, "INSERT_TEST_585960708904")
ida = IdaDataFrame(idadb, "INSERT_TEST_585960708904")
assert (ida.shape[0] == 0)
idadb._insert_into_database(df.loc[[1]], "INSERT_TEST_585960708904")
del ida.shape
assert (ida.shape[0] == 1)
idadb.drop_table("INSERT_TEST_585960708904")
def test_idadb_create_view(self, idadb, df):
try:
idadb.drop_table("CREATE_VIEW_TEST_585960708904")
except:
pass
idadb._create_table(df, "CREATE_VIEW_TEST_585960708904")
ida = IdaDataFrame(idadb, "CREATE_VIEW_TEST_585960708904")
try:
idadb.drop_view("VIEW_TEST_585960708904")
except:
pass
idadb._create_view(ida, "VIEW_TEST_585960708904")
idadb.drop_table("CREATE_VIEW_TEST_585960708904")
idadb.drop_view("VIEW_TEST_585960708904")
def apply_Model(X, Y):
# X_train=X_train.iloc[:,0:1139]
idadb = IdaDataBase("DASHDB")
idadf = IdaDataFrame(idadb)
IG = info_gain(idadb, target=Y, features=X)
def __init__(self, table_name='INTERSTATE_TRAFFIC'):
self.ida_traffic_pd = IdaDataFrame(idadb, table_name).as_dataframe()
def init_dataframe(self, tablename):
df = IdaDataFrame(self.idadb, tablename)
return df