Move tests into separate module.
Produce a graph of the magnetization.