fix uniqueness

Jammy2211 · Nov 11, 2024 · 4ed1a13 · 4ed1a13
1 parent b831b90
commit 4ed1a13
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/autoarray/structures/triangles/jax_coordinate_array.py b/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,

diff --git a/test_autoarray/structures/triangles/test_coordinate_jax.py b/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