Add unit test for whole simulation (work in progress)

Started implementing unit test for whole simulation. Not ready yet, but
doesn't effect building or running actually solution.