Add enable_restart docs and move tests to new class

[oliverwm1]

Closes #24

Adds documentation for enable_restart which was introduced in #9 without docs. Also improves organization of unit tests for enable_restart by adding them to a new TestClass and changing order of functions for clarity.