Update tests/locations_test.py

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

Author: domfournier

tests/locations_test.py

@@ -126,27 +126,22 @@ def test_filter(tmp_path: Path):
 #             ws,
 #             name="test-points",
 #             vertices=choices[inds],
-#         )
-#         tiles = tile_locations(pts.vertices[:, :2], n_tiles=8)
-#
-#         values = np.zeros(pts.n_vertices)
-#         pop = []
-#         for ind, tile in enumerate(tiles):
-#             values[tile] = ind
-#             pop.append(len(tile))
-#
-#         pts.add_data(
-#             {
-#                 "values": {
-#                     "values": values,
-#                 }
-#             }
-#         )
-#         assert np.std(pop) / np.mean(pop) < 0.02, (
-#             "Population of tiles are not almost equal {}."
-#         )
 
+def test_tile_locations():
+    n_points = 1000
+    rng = np.random.default_rng(0)
+    locations = rng.standard_normal((n_points, 2))
 
+    tiles = tile_locations(locations, n_tiles=8)
+
+    # All indices should be covered exactly once across tiles
+    all_indices = np.concatenate(tiles)
+    assert np.array_equal(np.sort(all_indices), np.arange(n_points))
+
+    # Tiles should be reasonably balanced in population
+    pop = np.array([len(tile) for tile in tiles])
+    assert pop.min() > 0
+    assert np.std(pop) / np.mean(pop) < 0.5
 def test_tile_locations_labels(tmp_path: Path):
     stn = np.arange(0, 10000, 1000)
     x_locs = np.kron(stn, np.ones(100)) + np.random.randn(1000) * 10