def test_changemanager(self):
cm = ChangeManager(buffer_updated=True,buffer_deleted=True,manage_columns=True)
self.assertEqual(cm.last_update(), 0)
self.assertEqual(cm.created_length(), 0)
self.assertEqual(cm.updated_length(), 0)
self.assertEqual(cm.deleted_length(), 0)
df = pd.DataFrame({'a': [ 1, 2, 3],
Module.UPDATE_COLUMN: [ 1, 1, 1 ]})
now = 1
cm.update(now, df)
self.assertEqual(cm.last_update(), now)
self.assertEqual(cm.next_created(),slice(0, 3))
self.assertEqual(cm.updated_length(), 0)
self.assertEqual(cm.deleted_length(), 0)
self.assertTrue(cm.column_changes.created.equals(df.columns))
self.assertTrue(cm.column_changes.kept.equals(NIL_INDEX))
self.assertTrue(cm.column_changes.deleted.equals(NIL_INDEX))
now = 2
df = df.append(pd.DataFrame({'a': [ 4], Module.UPDATE_COLUMN: [ now ]}),
ignore_index=True)
cm.update(now, df)
self.assertEqual(cm.last_update(), now)
self.assertEqual(cm.next_created(), slice(3,4))
self.assertEqual(cm.updated_length(), 0)
self.assertEqual(cm.deleted_length(), 0)
self.assertTrue(cm.column_changes.created.equals(NIL_INDEX))
self.assertTrue(cm.column_changes.kept.equals(df.columns))
self.assertTrue(cm.column_changes.deleted.equals(NIL_INDEX))
now = 3
df = df.append(pd.DataFrame({'a': [ 5], Module.UPDATE_COLUMN: [ now ]}),
ignore_index=True)
cm.update(now, df)
self.assertEqual(cm.last_update(), now)
self.assertEqual(cm.next_created(),slice(4, 5))
self.assertEqual(cm.updated_length(), 0)
self.assertEqual(cm.deleted_length(), 0)
self.assertTrue(cm.column_changes.created.equals(NIL_INDEX))
self.assertTrue(cm.column_changes.kept.equals(df.columns))
self.assertTrue(cm.column_changes.deleted.equals(NIL_INDEX))
now = 4
df2 = df[df.index != 2] # remove index==2
cm.update(now, df2)
self.assertEqual(cm.last_update(), now)
self.assertEqual(cm.created_length(), 0)
self.assertEqual(cm.updated_length(), 0)
self.assertEqual(cm.next_deleted(),slice(2,3))
self.assertTrue(cm.column_changes.created.equals(NIL_INDEX))
self.assertTrue(cm.column_changes.kept.equals(df.columns))
self.assertTrue(cm.column_changes.deleted.equals(NIL_INDEX))