Move computation test to main class.