make binary weights checker more strict

Author: tyler-a-cox

hera_filters/dspec.py

@@ -117,8 +117,7 @@ def _are_wgts_binary(wgts):
     Returns:
         Value of True if wgts contains only 1's and 0's
     """
-    binary_total = np.sum(np.isclose(wgts, 0) + np.isclose(wgts, 1))
-    return binary_total == np.prod(wgts.shape)
+    return np.all((wgts == 0) | (wgts == 1))
 
 def place_data_on_uniform_grid(x, data, weights, xtol=1e-3):
     """If possible, place data on a uniformly spaced grid.

hera_filters/tests/test_dspec.py

@@ -78,6 +78,10 @@ def test_are_wgts_binary():
     wgts = np.random.uniform(0, 1, size=100)
     assert not dspec._are_wgts_binary(wgts)
 
+    # Binary weights array plus some small number should return False
+    wgts = np.random.choice([0, 1], size=100) + 1e-12
+    assert not dspec._are_wgts_binary(wgts)
+
 def test_delay_filter_2D():
     NCHAN = 128
     NTIMES = 10