Summary: Minor refactor of the union_find data structure. Testing added for union_find.
Test Plan: Manual
Reviewers: sale, buda
Reviewed By: sale, buda
Subscribers: pullbot, florijan, sale, buda
Differential Revision: https://phabricator.memgraph.io/D42
