def test_aggregate_cell_data_df_shape_no_keep_cols(self):
dfs = [
pd.DataFrame(
{
'a': [0, 1, 2],
'b': [1, 2, 3]
}
),
pd.DataFrame(
{
'a': [0, 1, 2],
'b': [2, 3, 4]
}
)
]
dfs_merged = aggregate_cell_data(dfs)
self.assertCountEqual(dfs_merged.shape, [6, 2])
def test_aggregate_cell_data_col_names_no_keep_cols(self):
dfs = [
pd.DataFrame(
{
'a': [0, 1, 2],
'b': [1, 2, 3]
}
),
pd.DataFrame(
{
'a': [0, 1, 2],
'b': [2, 3, 4]
}
)
]
dfs_merged = aggregate_cell_data(dfs)
self.assertCountEqual(dfs_merged.columns, ['a', 'b'])
def test_aggregate_cell_data_col_names_cell_ids(self):
dfs = [
pd.DataFrame(
{
'a': [0, 1, 2],
'b': [1, 2, 3]
}
),
pd.DataFrame(
{
'a': [0, 1, 2],
'b': [2, 3, 4]
}
)
]
dfs_merged = aggregate_cell_data(
dfs,
cell_ids = ['001', '000'],
keep_cols = ['b']
)
self.assertCountEqual(dfs_merged.columns, ['a', '001_b', '000_b'])