fix uniqueness

Author: rhayes777

autoarray/structures/triangles/jax_coordinate_array.py

@@ -130,6 +130,7 @@ def neighborhood(self) -> "CoordinateArrayTriangles":
                 new_coordinates,
                 axis=0,
                 size=4 * self.coordinates.shape[0],
+                fill_value=np.nan,
             ),
             side_length=self.side_length,
             flipped=self.flipped,

test_autoarray/structures/triangles/test_coordinate_jax.py

@@ -103,3 +103,11 @@ def test_up_sample(one_triangle):
             ]
         ),
     )
+
+
+def test_means(one_triangle):
+    assert len(one_triangle.means) == 1
+
+    up_sampled = one_triangle.up_sample()
+    neighborhood = up_sampled.neighborhood()
+    assert np.count_nonzero(~np.isnan(neighborhood.means)) == 20