Add tests for UnionFind
Move tests out of package, add imports so they work.