WIP: Add unit tests for edit_handlers.py. Lint edit_handlers.py.

It looks like ```assertIn``` might be a better fit here