Rewrite tests in ginkgo

Much nicer than it was before