def test_iter_groups(self):
dfs = [
DataFrame({
f'df_{i}_column_0': np.ones(5) * i,
f'df_{i}_column_1': np.arange(5) * i
}) for i in range(5)
]
mdf = MDF()
for df in dfs:
mdf.append(df)
for i, mdf_df in enumerate(mdf.iter_groups()):
self.assertTrue(mdf_df.equals(dfs[i]))
