def test_remove_duplicates_direct(self): nodes = defaultdict(list) nodes["cube1_cluster0"] = [0] nodes["cube1_cluster1"] = [1, 2] nodes["cube1_cluster2"] = [1, 2, 3] nodes["cube2_cluster0"] = [1, 2] nodes["cube2_cluster1"] = [2, 3, 4] mapper = KeplerMapper() deduped_nodes = mapper._remove_duplicate_nodes(nodes) assert len(deduped_nodes) < len(nodes) assert len(deduped_nodes) == 4 assert ("cube1_cluster1|cube2_cluster0" in deduped_nodes or "cube2_cluster0|cube1_cluster1" in deduped_nodes)