umap
umap copied to clipboard
Fix flaky test
This is a fix for the flaky test discussed in #731. We found that updating the two assertions in the test reduces the flakiness to ~0%.
Please let me know if this change seems reasonable.