Move everything into test unit.

Why? Because I'm too lazy to write comment on what are the expected results.